diff --git a/gpu-cuda/Dockerfile b/gpu-cuda/Dockerfile
index 937f6cb..1cf95d5 100644
--- a/gpu-cuda/Dockerfile
+++ b/gpu-cuda/Dockerfile
@@ -20,5 +20,7 @@ USER ${NB_UID}
 COPY --chown=${NB_UID}:${NB_GID} requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
+RUN pip install --force-reinstall --no-cache-dir -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ "jupyter-collaboration-codio==4.1.4rc0"
+
 # Expose Jupyter port
 EXPOSE 8888
diff --git a/gpu-cuda/requirements.txt b/gpu-cuda/requirements.txt
index d2a5d88..0a606b4 100644
--- a/gpu-cuda/requirements.txt
+++ b/gpu-cuda/requirements.txt
@@ -35,6 +35,7 @@ einops
 # Jupyter Extensions
 ipywidgets
 jupyter-contrib-nbextensions
+jupyterlab-v4-codio-ext
 
 # Other Utilities
 Pillow
diff --git a/standard/Dockerfile b/standard/Dockerfile
index 9189b49..b83303d 100644
--- a/standard/Dockerfile
+++ b/standard/Dockerfile
@@ -16,5 +16,7 @@ USER ${NB_UID}
 COPY --chown=${NB_UID}:${NB_GID} requirements.txt /tmp/
 RUN pip install --no-cache-dir -r /tmp/requirements.txt
 
+RUN pip install --force-reinstall --no-cache-dir -i https://test.pypi.org/simple/ --extra-index-url https://pypi.org/simple/ "jupyter-collaboration-codio==4.1.4rc0"
+
 # Expose Jupyter port
 EXPOSE 8888
diff --git a/standard/requirements.txt b/standard/requirements.txt
index 33991ba..34cf3fc 100644
--- a/standard/requirements.txt
+++ b/standard/requirements.txt
@@ -18,6 +18,7 @@ einops
 # Jupyter Extensions
 ipywidgets
 jupyter-contrib-nbextensions
+jupyterlab-v4-codio-ext
 
 # Other Utilities
 Pillow