diff --git a/python/requirements/ml/requirements_dl.txt b/python/requirements/ml/requirements_dl.txt
index 3970fa10a975..50a021869c93 100644
--- a/python/requirements/ml/requirements_dl.txt
+++ b/python/requirements/ml/requirements_dl.txt
@@ -7,7 +7,7 @@
 
 # TODO(amogkam): Remove after https://github.com/tensorflow/tensorflow/issues/52922 is fixed.
 keras==2.6.0
-tensorflow==2.6.0
+tensorflow==2.6.1
 tensorflow-probability==0.14.0
 
 torch==1.9.0;sys_platform=="darwin"