Merge pull request #2 from leiserfg/fix-single-slot

Fix ask_slot when there is a single slot
not-a-feature · Apr 30, 2024 · adb6080 · adb6080
2 parents 56b8360 + d9f0c06
commit adb6080
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/wd_fw_update/main.py b/src/wd_fw_update/main.py
@@ -259,7 +259,8 @@ def ask_slot(device: str) -> Tuple[int, int]:
     if len(slots) == 1:
         _logger.info("Only one slot to select, skipping user-promt.")
 
-        return int(slots[0].split(":")[0])  # Should always be 1
+        slot = int(slots[0].split(":")[0])  # Should always be 1
+        return current_slot, slot
 
     print("Select the slot to which the firmware should be installed.")
     questions = [