diff --git a/gpu-cuda/Dockerfile b/gpu-cuda/Dockerfile
index 1cf95d5..a813125 100644
--- a/gpu-cuda/Dockerfile
+++ b/gpu-cuda/Dockerfile
@@ -20,6 +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/ "jupyterlab-v4-codio-ext==0.1.22"
 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
diff --git a/gpu-cuda/requirements.txt b/gpu-cuda/requirements.txt
index 0a606b4..d2a5d88 100644
--- a/gpu-cuda/requirements.txt
+++ b/gpu-cuda/requirements.txt
@@ -35,7 +35,6 @@ einops
 # Jupyter Extensions
 ipywidgets
 jupyter-contrib-nbextensions
-jupyterlab-v4-codio-ext
 
 # Other Utilities
 Pillow
diff --git a/standard/Dockerfile b/standard/Dockerfile
index b83303d..84a64f8 100644
--- a/standard/Dockerfile
+++ b/standard/Dockerfile
@@ -16,6 +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/ "jupyterlab-v4-codio-ext==0.1.22"
 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
diff --git a/standard/requirements.txt b/standard/requirements.txt
index 34cf3fc..33991ba 100644
--- a/standard/requirements.txt
+++ b/standard/requirements.txt
@@ -18,7 +18,6 @@ einops
 # Jupyter Extensions
 ipywidgets
 jupyter-contrib-nbextensions
-jupyterlab-v4-codio-ext
 
 # Other Utilities
 Pillow