diff --git a/compile.sh b/compile.sh
index 32f9ffa6b..25973bdf1 100755
--- a/compile.sh
+++ b/compile.sh
@@ -288,9 +288,6 @@ while [[ "${1}" == *=* ]]; do
 
 done
 
-# Used to prepare Docker image
-[[ "${JUST_INIT}" == "yes" ]] && exit 0
-
 
 if [[ "${BUILD_ALL}" == "yes" || "${BUILD_ALL}" == "demo" ]]; then
 
diff --git a/lib/main.sh b/lib/main.sh
index a4ff1857c..630523c54 100644
--- a/lib/main.sh
+++ b/lib/main.sh
@@ -428,6 +428,8 @@ start=$(date +%s)
 # The OFFLINE_WORK variable inside the function
 prepare_host
 
+[[ "${JUST_INIT}" == "yes" ]] && exit 0
+
 [[ $CLEAN_LEVEL == *sources* ]] && cleaning "sources"
 
 # fetch_from_repo <url> <dir> <ref> <subdir_flag>