diff --git a/scripts/Muon/maxent_model.py b/scripts/Muon/maxent_model.py
index 8a522df6bf9c05da3fcf557a2f415e74bed3b727..98f4b20ebfbabc726276c0f80689de827126a865 100644
--- a/scripts/Muon/maxent_model.py
+++ b/scripts/Muon/maxent_model.py
@@ -21,14 +21,8 @@ class MaxEntWrapper(object):
         """
         store the data in the wrapper for later
         """
-        if "phaseTable" in inputs:
-            self.phaseTable = inputs["phaseTable"]
-        else:
-            self.phaseTable = None
-        if "maxent" in inputs:
-            self.maxent = inputs["maxent"]
-        else:
-            self.maxent = None
+        self.phaseTable = inputs.get("phaseTable",None)
+        self.maxent = inputs.get("maxent",None)
         self.model.setRun(inputs["Run"])
 
     def execute(self):