diff --git a/stage0/00-configure-apt/00-run.sh b/stage0/00-configure-apt/00-run.sh
index 8d143ed..f978814 100755
--- a/stage0/00-configure-apt/00-run.sh
+++ b/stage0/00-configure-apt/00-run.sh
@@ -14,11 +14,4 @@ on_chroot apt-key add - < files/raspberrypi.gpg.key
 on_chroot << EOF
 apt-get update
 apt-get dist-upgrade -y
-EOF
-
-on_chroot << EOF
-wget -q "https://raw.githubusercontent.com/moby/moby/master/contrib/download-frozen-image-v2.sh"
-chmod +x download-frozen-image-v2.sh
-./download-frozen-image-v2.sh docker-images getumbrel/dashboard:latest
-tar -cC 'target_dir' . | docker load
 EOF
\ No newline at end of file
diff --git a/stage2/05-docker-compose/01-run.sh b/stage2/05-docker-compose/01-run.sh
index 0e6b389..5d99a2c 100755
--- a/stage2/05-docker-compose/01-run.sh
+++ b/stage2/05-docker-compose/01-run.sh
@@ -30,3 +30,11 @@ cp files/compose-service ${ROOTFS_DIR}/etc/init.d/umbrelbox
 
 
 echo "Docker stuff installed!"
+
+echo "Downloading images from Docker Hub"
+on_chroot << EOF
+wget -q "https://raw.githubusercontent.com/moby/moby/master/contrib/download-frozen-image-v2.sh"
+chmod +x download-frozen-image-v2.sh
+./download-frozen-image-v2.sh docker-images getumbrel/middleware:latest
+tar -cC 'target_dir' . | docker load
+EOF
\ No newline at end of file