diff --git a/aiserver.py b/aiserver.py
index 29a9a3d1..e0097572 100644
--- a/aiserver.py
+++ b/aiserver.py
@@ -234,6 +234,7 @@ def device_config(model):
                 primaryselect = input("device ID> ")
                 if(primaryselect.isnumeric() and 0 <= int(primaryselect) < device_count):
                     breakmodel.primary_device = int(primaryselect)
+                    break
                 else:
                     print(f"{colors.RED}Please enter an integer between 0 and {device_count-1}.{colors.END}")
         else: