diff --git a/scripts/cdist b/scripts/cdist
index d02f0a5f..5f8fff9f 100755
--- a/scripts/cdist
+++ b/scripts/cdist
@@ -49,9 +49,10 @@ def check_beta(args_dict):
     # raise error.
     if not args_dict['beta']:
         cmd = args_dict['command']
-        for arg in BETA_ARGS[cmd]:
-            if arg in args_dict:
-                raise cdist.CdistBetaRequired(cmd, arg)
+        if cmd in BETA_ARGS:
+            for arg in BETA_ARGS[cmd]:
+                if arg in args_dict:
+                    raise cdist.CdistBetaRequired(cmd, arg)
 
 
 def commandline():