[client] Fix marking definition addition in entities (#140)

OpenCTI-Platform · Feb 14, 2021 · 4217fcd · 4217fcd
1 parent b469030
commit 4217fcd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/pycti/entities/opencti_stix_core_relationship.py b/pycti/entities/opencti_stix_core_relationship.py
@@ -589,7 +589,7 @@ def add_marking_definition(self, **kwargs):
                     "error", "Cannot add Marking-Definition, entity not found"
                 )
                 return False
-            if marking_definition_id in stix_core_relationship["markingDefinitionsIds"]:
+            if marking_definition_id in stix_core_relationship["objectMarkingIds"]:
                 return True
             else:
                 self.opencti.log(

diff --git a/pycti/entities/opencti_stix_cyber_observable.py b/pycti/entities/opencti_stix_cyber_observable.py
@@ -1250,7 +1250,7 @@ def add_marking_definition(self, **kwargs):
                     "error", "Cannot add Marking-Definition, entity not found"
                 )
                 return False
-            if marking_definition_id in stix_cyber_observable["markingDefinitionsIds"]:
+            if marking_definition_id in stix_cyber_observable["objectMarkingIds"]:
                 return True
             else:
                 self.opencti.log(

diff --git a/pycti/entities/opencti_stix_domain_object.py b/pycti/entities/opencti_stix_domain_object.py
@@ -839,7 +839,7 @@ def add_marking_definition(self, **kwargs):
                     "error", "Cannot add Marking-Definition, entity not found"
                 )
                 return False
-            if marking_definition_id in stix_domain_object["markingDefinitionsIds"]:
+            if marking_definition_id in stix_domain_object["objectMarkingIds"]:
                 return True
             else:
                 self.opencti.log(