diff --git a/Justfile b/Justfile
index 0e67776..b69e4fa 100644
--- a/Justfile
+++ b/Justfile
@@ -5,6 +5,10 @@ dev +command:
 
 deploy build="":
     git pull
-    docker compose up {{ if build != "" { "--build" } else { "" } }} --pull=always -d
-    {{if build == ""}}docker compose restart website{{end}}
+    # Set variable for build option
+    {{ set build_option = if build != "" { "--build" } else { "" } }}
+    docker compose up {{ build_option }} --pull=always -d
+    # Set variable for restart command
+    {{ set restart_cmd = if build == "" { "docker compose restart website" } else { "" } }}
+    {{ restart_cmd }}
     docker system prune -f
\ No newline at end of file