diff --git a/imitation_learning/UCB_DRL/CW02/requirements.txt b/imitation_learning/UCB_DRL/CW02/requirements.txt
index ad3a498..1dbdda9 100644
--- a/imitation_learning/UCB_DRL/CW02/requirements.txt
+++ b/imitation_learning/UCB_DRL/CW02/requirements.txt
@@ -1,6 +1,6 @@
 mujoco-py==1.50.1.56
 gym==0.10.5
-tensorflow==2.5.0
+tensorflow==2.11.1
 numpy==1.14.5
 seaborn
 Box2D==2.3.2