diff --git a/docker/Dockerfile b/docker/Dockerfile
index aeaa07f..67e9bac 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -2,7 +2,7 @@
 FROM cernml4reco/djcbase:py2
 
 RUN export DEBIAN_FRONTEND=noninteractive && \
-    pip --no-cache-dir install tensorflow-gpu==2.1
+    pip --no-cache-dir install tensorflow-gpu==1.15
     
 RUN cd /usr/local/lib/python2.7/dist-packages/tensorflow_core/include/third_party && \
     mkdir gpus && \
@@ -34,7 +34,7 @@ RUN cd /usr/share && \
     cd DJC && \
     git clone https://github.com/DL4Jets/DeepJetCore && \
     cd DeepJetCore && \
-    git checkout 2.2_tf2.1 &&\
+    git checkout 2.2_tf1.15 &&\
     source docker_env.sh && \
     cd compiled && \
     make -j4