diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 8a6663034374abbb13231e33b87f27519a68257e..820bd08f8816e6f8b493aa4019740e6d7dc5ba20 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -28,6 +28,8 @@ stages:
     GIT_SUBMODULE_DEPTH: 1
     GIT_SUBMODULE_UPDATE_FLAGS: --jobs 4
   before_script:
+    - git lfs pull -I mattergen/checkpoints/chemical_system --exclude=""
+    - git lfs pull -I mattergen/checkpoints/chemical_system_energy_above_hull --exclude=""
     - echo "{\"auths\":{\"${CI_REGISTRY}\":{\"auth\":\"$(printf "%s:%s" "${CI_REGISTRY_USER}" "${CI_REGISTRY_PASSWORD}" | base64 | tr -d '\n')\"},\"$CI_DEPENDENCY_PROXY_SERVER\":{\"auth\":\"$(printf "%s:%s" ${CI_DEPENDENCY_PROXY_USER} "${CI_DEPENDENCY_PROXY_PASSWORD}" | base64 | tr -d '\n')\"}}}" > /kaniko/.docker/config.json
 
 build:
diff --git a/Dockerfile b/Dockerfile
index 007200e1a735e62bc58553a0417f0160328cb917..c1b8d8a35c469d0d9ebbca72e8e3023e636e0a87 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -4,15 +4,15 @@ FROM quay.io/jupyter/minimal-notebook:lab-4.2.5
 # Fix: https://github.com/koalaman/shellcheck/wiki/SC3014
 SHELL ["/bin/bash", "-o", "pipefail", "-c"]
 
-USER root
-
-RUN apt-get update --yes \
- && apt-get install --yes --no-install-recommends \
-    git-lfs \
- && apt-get clean \
- && rm -rf /var/lib/apt/lists/*
-
-USER ${NB_UID}
+# USER root
+#
+# RUN apt-get update --yes \
+#  && apt-get install --yes --no-install-recommends \
+#     git-lfs \
+#  && apt-get clean \
+#  && rm -rf /var/lib/apt/lists/*
+# 
+# USER ${NB_UID}
 
 # RUN mamba install --yes -c pytorch -c nvidia \
 #     pytorch==2.2.1 \