diff --git a/docker/install/ubuntu_install_tensorflow.sh b/docker/install/ubuntu_install_tensorflow.sh
index eaf89ffcf8fe..1fa1cdaee014 100755
--- a/docker/install/ubuntu_install_tensorflow.sh
+++ b/docker/install/ubuntu_install_tensorflow.sh
@@ -21,6 +21,6 @@ set -u
 set -o pipefail
 
 pip3 install \
-    "h5py==3.1.0" \
+    h5py==3.1.0 \
     keras==2.6 \
-    tensorflow==2.6.2
+    tensorflow==2.6.5