diff --git a/Dockerfile b/Dockerfile
index 7fc4875..4dae941 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -55,6 +55,9 @@ ENV EMAIL_ADDRESS="youremail@example.com"
 ENV GPG_PASSWORD="PickAPassword"
 ENV HOSTNAME=localhost
 
+ARG MODE=packages
+ENV MODE ${MODE}
+
 COPY assets/packages /opt/packages
 
 RUN /opt/startup.sh
diff --git a/assets/update_mirror_ubuntu.sh b/assets/update_mirror_ubuntu.sh
index 330344f..72d8212 100755
--- a/assets/update_mirror_ubuntu.sh
+++ b/assets/update_mirror_ubuntu.sh
@@ -15,7 +15,7 @@ UBUNTU_RELEASE=xenial
 UPSTREAM_URL="http://archive.ubuntu.com/ubuntu/"
 COMPONENTS=( main universe )
 REPOS=( ${UBUNTU_RELEASE} ${UBUNTU_RELEASE}-updates ${UBUNTU_RELEASE}-security )
-MODE='packages' # packages - mirror specified packages or all
+
 if [ "$MODE" = "packages" ]; then
     FILTER_OPTS=(-filter="$(cat /opt/packages | paste -sd \| -)" -filter-with-deps)
 else