diff --git a/Surrogate_MBO/requirements.txt b/Surrogate_MBO/requirements.txt
index 3b02b4b..04208e9 100644
--- a/Surrogate_MBO/requirements.txt
+++ b/Surrogate_MBO/requirements.txt
@@ -124,7 +124,7 @@ tabulate==0.8.7
 tensorboard==2.2.1
 tensorboard-plugin-wit==1.6.0.post3
 tensorboardX==2.0
-tensorflow==2.4.0
+tensorflow==2.5.2
 tensorflow-estimator==2.2.0
 tensorflow-probability==0.10.0
 termcolor==1.1.0