diff --git a/org.ietr.preesm.reinforcement_learning/Ref/Python-Version/requirements.txt b/org.ietr.preesm.reinforcement_learning/Ref/Python-Version/requirements.txt
index 97db814c..9b7dd35f 100644
--- a/org.ietr.preesm.reinforcement_learning/Ref/Python-Version/requirements.txt
+++ b/org.ietr.preesm.reinforcement_learning/Ref/Python-Version/requirements.txt
@@ -1,3 +1,3 @@
-tensorflow==1.4.0
+tensorflow==2.6.4
 numpy==1.13.3
 gym==0.9.4