diff --git a/Dockerfile b/Dockerfile
index a380c70..3cb28c0 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -11,7 +11,7 @@ WORKDIR $CODE_DIR/baselines
 # Clean up pycache and pyc files
 RUN rm -rf __pycache__ && \
     find . -name "*.pyc" -delete && \
-    pip install tensorflow<2 && \
+    pip install 'tensorflow < 2' && \
     pip install -e .[test]