From 5f825d1da9c1dcc3fa50f4df1ef5a7860453e3e5 Mon Sep 17 00:00:00 2001
From: Darko Poljak <foss@ungleich.com>
Date: Mon, 15 Aug 2016 18:04:57 +0200
Subject: [PATCH] Fix check_beta error if cmd is not in beta args dict.

---
 scripts/cdist | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

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():