diff --git a/scripts/helpers/prompt.py b/scripts/helpers/prompt.py
index 48dc46c..1d41b5a 100755
--- a/scripts/helpers/prompt.py
+++ b/scripts/helpers/prompt.py
@@ -45,7 +45,8 @@ def debug(ctx):
 @click.argument('order')
 @click.argument('format')
 @click.pass_context
-def set(ctx, order: int, format: str):
+def set(ctx, order: str, format: str):
+    order = int(order)
     parts = ctx.obj['parts']
 
     # Create new parts list