diff --git a/Dockerfile b/Dockerfile
index e97df3e..f905556 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,6 +15,8 @@ RUN apt update -y && \
     apt update -y && \
     apt install graphviz -y
 
+RUN jupyter labextension install @dafeliton/jupyterlab-notebookparams
+
 # Install geospatial packages first
 RUN pip uninstall pillow fiona -y && \
     pip install -r ~/requirements.txt && \
@@ -35,8 +37,6 @@ RUN mamba install -c esri arcgis arcgis-mapping -y && \
 RUN pip install --upgrade pip setuptools wheel
 RUN pip install --upgrade nbconvert
 
-RUN jupyter labextension install @dafeliton/jupyterlab-notebookparams --no-build
-
 
 # RUN pip install "numpy<2"