diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml
index 820bd08f8816e6f8b493aa4019740e6d7dc5ba20..8a6663034374abbb13231e33b87f27519a68257e 100644
--- a/.gitlab-ci.yml
+++ b/.gitlab-ci.yml
@@ -28,8 +28,6 @@ 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 c1b8d8a35c469d0d9ebbca72e8e3023e636e0a87..007200e1a735e62bc58553a0417f0160328cb917 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 \