diff --git a/setup.sh b/setup.sh
index 538bbc8..c6448cb 100755
--- a/setup.sh
+++ b/setup.sh
@@ -50,6 +50,7 @@ source "$REPO/steps/step_onyx.sh"
 source "$REPO/steps/step_models.sh"
 source "$REPO/steps/step_webconfig.sh"
 source "$REPO/steps/step_rag.sh"
+source "$REPO/steps/step_sleep.sh"
 
 # ---------------------------------------------------------------------------
 # Main setup
@@ -73,6 +74,7 @@ install_homebrew
 install_ollama
 install_docker
 install_onyx
+prevent_sleep
 setup_models
 web_config "host.docker.internal"
 rag_upload
diff --git a/setup_ollama.sh b/setup_ollama.sh
index 428c31b..87d2057 100755
--- a/setup_ollama.sh
+++ b/setup_ollama.sh
@@ -45,6 +45,7 @@ source "$REPO/steps/preflight.sh"
 source "$REPO/steps/step_homebrew.sh"
 source "$REPO/steps/step_ollama.sh"
 source "$REPO/steps/step_models.sh"
+source "$REPO/steps/step_sleep.sh"
 
 # ---------------------------------------------------------------------------
 # Main setup
@@ -66,6 +67,7 @@ preflight_checks --skip-docker
 install_homebrew
 install_ollama
 setup_models
+prevent_sleep
 
 # =============================================================================
 # Done
diff --git a/setup_onyx.sh b/setup_onyx.sh
index 8b217ab..836a5a4 100755
--- a/setup_onyx.sh
+++ b/setup_onyx.sh
@@ -64,6 +64,7 @@ source "$REPO/steps/step_docker.sh"
 source "$REPO/steps/step_onyx.sh"
 source "$REPO/steps/step_webconfig.sh"
 source "$REPO/steps/step_rag.sh"
+source "$REPO/steps/step_sleep.sh"
 
 # ---------------------------------------------------------------------------
 # Detect where Ollama is running
@@ -154,6 +155,7 @@ preflight_checks
 install_homebrew
 install_docker
 install_onyx
+prevent_sleep
 
 # Resolve Ollama host after Docker is running so host.docker.internal works
 if [ -n "$OLLAMA_HOST_OVERRIDE" ]; then
diff --git a/steps/step_onyx.sh b/steps/step_onyx.sh
index 3c343ad..a46fe67 100755
--- a/steps/step_onyx.sh
+++ b/steps/step_onyx.sh
@@ -20,8 +20,4 @@ install_onyx() {
 
         print_success "Onyx installation complete"
     fi
-
-    print_warning "Configuring power settings to prevent sleep..."
-    sudo pmset -a sleep 0 disksleep 0
-    print_success "Sleep prevention configured"
 }
diff --git a/steps/step_sleep.sh b/steps/step_sleep.sh
new file mode 100755
index 0000000..fe50ead
--- /dev/null
+++ b/steps/step_sleep.sh
@@ -0,0 +1,10 @@
+#!/bin/bash
+# Step: Prevent the machine from sleeping. Source common.sh before sourcing this file.
+
+prevent_sleep() {
+    print_step "Preventing Sleep"
+
+    print_warning "Configuring power settings to prevent sleep..."
+    sudo pmset -a sleep 0 disksleep 0
+    print_success "Sleep prevention configured"
+}