fix: Properly pop the distribution key.

MISP · Oct 26, 2017 · 14bc9e4 · 14bc9e4
1 parent 78eb6e3
commit 14bc9e4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pymisp/mispevent.py b/pymisp/mispevent.py
@@ -177,8 +177,9 @@ def from_dict(self, **kwargs):
         if not isinstance(self.to_ids, bool):
             raise NewAttributeError('{} is invalid, to_ids has to be True or False'.format(self.to_ids))
 
-        if kwargs.get('distribution') is not None:
-            self.distribution = int(kwargs.pop('distribution'))
+        self.distribution = kwargs.pop('distribution', None)
+        if self.distribution is not None:
+            self.distribution = int(self.distribution)
             if self.distribution not in [0, 1, 2, 3, 4, 5]:
                 raise NewAttributeError('{} is invalid, the distribution has to be in 0, 1, 2, 3, 4, 5'.format(self.distribution))