diff --git a/app/atomic_svc.py b/app/atomic_svc.py index 38e3690b..d7f43db6 100644 --- a/app/atomic_svc.py +++ b/app/atomic_svc.py @@ -301,7 +301,7 @@ async def _save_ability(self, entries, test): """ Return True if an ability was saved. """ - ability_id = hashlib.md5(json.dumps(test).encode()).hexdigest() + ability_id = test.get('auto_generated_guid') or hashlib.md5(json.dumps(test).encode()).hexdigest() tactics_li = self.technique_to_tactics.get(entries['attack_technique'], ['redcanary-unknown']) tactic = 'multiple' if len(tactics_li) > 1 else tactics_li[0]