diff --git a/tf-nonlinearity/requirements.txt b/tf-nonlinearity/requirements.txt
index 1f6f1ec..b49b37a 100644
--- a/tf-nonlinearity/requirements.txt
+++ b/tf-nonlinearity/requirements.txt
@@ -4,5 +4,5 @@ ipython==7.16.3
 matplotlib==1.5.3
 numpy==1.21.0
 scipy==0.18.1
-tensorflow==2.5.3
+tensorflow==2.6.4
 tqdm==4.8.4