[Bug] Flatten method improperly added subtechniques (#1404)

2021-08-05 11:15:07 -08:00
parent 17bf3c1e16
commit 95486ecfdf
1 changed files with 2 additions and 2 deletions
@@ -103,8 +103,8 @@ class ThreatMapping(MarshmallowDataclassMixin):
                technique_ids.add(technique.id)

                for subtechnique in (technique.subtechnique or []):
-                    sub_technique_ids.update(subtechnique.id)
-                    sub_technique_names.update(subtechnique.name)
+                    sub_technique_ids.add(subtechnique.id)
+                    sub_technique_names.add(subtechnique.name)

        return FlatThreatMapping(
            tactic_names=sorted(tactic_names),