Skip to content

Commit 822d85d

Browse files
committed
fix MatrixGen init method for local sources
1 parent 90a3615 commit 822d85d

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mitreattack/navlayers/exporters/matrix_gen.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -142,11 +142,11 @@ def __init__(self, source="local", resource=None, domain="enterprise"):
142142
if resource is not None:
143143
hd = MemoryStore()
144144
hd.load_from_file(resource)
145-
if "mobile" in resource.lower():
145+
if "mobile" in domain.lower():
146146
self.collections["mobile"] = hd
147-
elif "enterprise" in resource.lower():
147+
elif "enterprise" in domain.lower():
148148
self.collections["enterprise"] = hd
149-
elif "ics" in resource.lower():
149+
elif "ics" in domain.lower():
150150
self.collections["ics"] = hd
151151
else:
152152
logger.error(f"invalid domain specified ({resource.lower()})")

0 commit comments

Comments
 (0)