diff --git a/setup.py b/setup.py
index 755beac3..c032fccd 100644
--- a/setup.py
+++ b/setup.py
@@ -49,7 +49,7 @@
     'scikit-learn==0.24.2',
     'scipy==1.4.0',
     'six==1.15.0',
-    'tensorflow==2.5.0',
+    'tensorflow==2.6.4',
     'tqdm==4.61.1',
     'umap-learn==0.5.1',
     'yale-dhlab-rasterfairy>=1.0.3',