diff --git a/datacenterlight/management/commands/fix_vm_after_celery_error.py b/datacenterlight/management/commands/fix_vm_after_celery_error.py
index 97df553e..2f7325f4 100644
--- a/datacenterlight/management/commands/fix_vm_after_celery_error.py
+++ b/datacenterlight/management/commands/fix_vm_after_celery_error.py
@@ -13,7 +13,9 @@ class Command(BaseCommand):
     def add_arguments(self, parser):
         parser.add_argument('vm_id', type=int)
         parser.add_argument('order_id', type=int)
+        parser.add_argument('user', type=str)
         parser.add_argument('specs', type=str)
+        parser.add_argument('template', type=str)
 
     def handle(self, *args, **options):
         vm_id = options['vm_id']