diff --git a/ollama/docker-compose.yml b/ollama/docker-compose.yml
index 22dac33..1a22c88 100755
--- a/ollama/docker-compose.yml
+++ b/ollama/docker-compose.yml
@@ -33,6 +33,7 @@ services:
       # Disables user/password login form. Required when ENABLE_OAUTH_SIGNUP=true.
       - ENABLE_LOGIN_FORM=false
       - OAUTH_MERGE_ACCOUNTS_BY_EMAIL=true
+    restart: ${RESTART:-unless-stopped}
 volumes:
   open-webui:
   models:
\ No newline at end of file