diff --git a/gpu.dockerfile b/gpu.dockerfile
index e0f43c9..6775966 100644
--- a/gpu.dockerfile
+++ b/gpu.dockerfile
@@ -28,8 +28,8 @@ ENV CONDA_DIR=${CONDA_DIR:-/opt/conda} \
     LANG=${LANG:-en_US.UTF-8} \
     LANGUAGE=${LANGUAGE:-en_US.UTF-8}
 ENV PATH="${CONDA_DIR}/bin:${PATH}"
-COPY install_conda.sh .
-RUN ./install_conda.sh
+COPY install_conda.sh /tmp/
+RUN /tmp/install_conda.sh
 
 ## Install packages with Conda
 RUN conda install --quiet -y \
@@ -37,7 +37,7 @@ RUN conda install --quiet -y \
       nodejs \
       ipywidgets \
       nb_conda_kernels \
-    #   jupyterlab \
+      jupyterlab \
       jupyterlab-git \
       jupyterlab-lsp \
       jupyter-lsp-python \
diff --git a/jupyter_notebook_config.py b/jupyter_notebook_config.py
index 1a69c52..0225bef 100644
--- a/jupyter_notebook_config.py
+++ b/jupyter_notebook_config.py
@@ -32,10 +32,12 @@
     os.system('cat extensions.txt | xargs -I {} jupyter {} install --user')
 
 if os.path.exists('environment.yml'):
-    os.system('mamba env create -f environment.yml')
+    os.system('conda env create -f environment.yml')
+    # os.system('mamba env create -f environment.yml')
 
 if os.path.exists('environment.yaml'):
-    os.system('mamba env create -f environment.yaml')
+    os.system('conda env create -f environment.yaml')
+    # os.system('mamba env create -f environment.yaml')
 
 if workspace:
     os.chdir(workspace)