diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 2dcb0ccc786e91699ad214036f1271a48aba37bd..a627c5cbc8098f0a6e6840ee5b776f56db2bef0b 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -9,5 +9,5 @@ docker-build-master:
   before_script:
     - docker login -u "$CI_REGISTRY_USER" -p "$CI_REGISTRY_PASSWORD" $CI_REGISTRY
   script:
-    - docker build --pull -t "$CI_REGISTRY_IMAGE":prod .
-    - docker push "$CI_REGISTRY_IMAGE":prod
+    - docker build --pull -t "$CI_REGISTRY_IMAGE":prodcc4 .
+    - docker push "$CI_REGISTRY_IMAGE":prodcc4
diff --git a/Dockerfile b/Dockerfile
index 1de2bf398c7674556d5f3730b357640dd2bc578b..43d97b0fe07ce25c9e3c4976eaae761b0bf60e61 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM jupyter/tensorflow-notebook:hub-1.5.0
+FROM quay.io/jupyter/tensorflow-notebook:hub-5.2.0
 
 USER root
 
@@ -7,6 +7,7 @@ RUN pip3 install --upgrade pip
 RUN mamba update mamba
 
 # Conda packages
+RUN mamba install -c conda-forge jupyter-server-proxy==4.4.0
 RUN mamba install -c conda-forge control
 RUN mamba install -c conda-forge casadi
 RUN mamba install -c conda-forge jax