Merge branch 'master' of github.com:oasis-open/cti-python-stix2

MISP · Jul 1, 2019 · b204b9f · b204b9f
2 parents 565acc7 + 244d8c9
commit b204b9f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/stix2/v20/bundle.py b/stix2/v20/bundle.py
@@ -43,7 +43,7 @@ def __init__(self, *args, **kwargs):
 
     def get_obj(self, obj_uuid):
         if "objects" in self._inner:
-            found_objs = [elem for elem in self.objects if elem.id == obj_uuid]
+            found_objs = [elem for elem in self.objects if elem['id'] == obj_uuid]
             if found_objs == []:
                 raise KeyError("'%s' does not match the id property of any of the bundle's objects" % obj_uuid)
             return found_objs

diff --git a/stix2/v21/bundle.py b/stix2/v21/bundle.py
@@ -41,7 +41,7 @@ def __init__(self, *args, **kwargs):
 
     def get_obj(self, obj_uuid):
         if "objects" in self._inner:
-            found_objs = [elem for elem in self.objects if elem.id == obj_uuid]
+            found_objs = [elem for elem in self.objects if elem['id'] == obj_uuid]
             if found_objs == []:
                 raise KeyError("'%s' does not match the id property of any of the bundle's objects" % obj_uuid)
             return found_objs