diff --git a/scripts/wk/clone/state.py b/scripts/wk/clone/state.py
index e9913d72..a5ae6325 100644
--- a/scripts/wk/clone/state.py
+++ b/scripts/wk/clone/state.py
@@ -1047,24 +1047,15 @@ def select_disk_obj(label:str, disk_menu: cli.Menu, disk_path: str) -> hw_disk.D
 
 def set_mode(docopt_args) -> str:
   """Set mode from docopt_args or user selection, returns str."""
-  mode = '?'
-
-  # Check docopt_args
   if docopt_args['clone']:
-    mode = 'Clone'
-  elif docopt_args['image']:
-    mode = 'Image'
+    return 'Clone'
+
+  if docopt_args['image']:
+    return 'Image'
 
   # Ask user if necessary
-  if not mode:
-    answer = cli.choice('Are we cloning or imaging?', ['C', 'I'])
-    if answer == 'C':
-      mode = 'Clone'
-    else:
-      mode = 'Image'
-
-  # Done
-  return mode
+  answer = cli.choice('Are we cloning or imaging?', ['C', 'I'])
+  return 'Clone' if answer == 'C' else 'Image'
 
 
 if __name__ == '__main__':