diff --git a/valohai_cli/commands/execution/run/frontend_command.py b/valohai_cli/commands/execution/run/frontend_command.py
index 549a03dd..57fa273b 100644
--- a/valohai_cli/commands/execution/run/frontend_command.py
+++ b/valohai_cli/commands/execution/run/frontend_command.py
@@ -111,8 +111,8 @@ def run(
                     f"does not seem valid (it should start with `ssh`)"
                 )
         runtime_config["debug_key"] = key
-    if autorestart_spot:
-        runtime_config["autorestart"] = autorestart_spot
+    if autorestart:
+        runtime_config["autorestart"] = autorestart
 
     rc = RunCommand(
         project=project,