diff --git a/scripts/install.sh b/scripts/install.sh
index 82eca656..dbc5ff6a 100644
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -92,7 +92,9 @@ EOF
             status "Enabling and starting ollama service..."
             $SUDO systemctl daemon-reload
             $SUDO systemctl enable ollama
-            $SUDO systemctl restart ollama
+
+            start_service() { $SUDO systemctl restart ollama; }
+            trap start_service EXIT
             ;;
     esac
 }