Merge pull request #9 from CybercentreCanada/assemblyline/issues/101

Bugfix
CybercentreCanada · Feb 18, 2024 · 95f8b70 · 95f8b70
2 parents 2cd0ef8 + 92433f8
commit 95f8b70
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/badlist/badlist.py b/badlist/badlist.py
@@ -123,7 +123,7 @@ def execute(self, request):
                         "Last updated": badlisted["updated"],
                     },
                     heuristic=Heuristic(2, score_map=self.source_score_override, signatures=badlisted["sources"]),
-                    classification=source.get("classification", classification.UNRESTRICTED),
+                    classification=badlisted.get("classification", classification.UNRESTRICTED),
                     tags={badlisted["tag"]["type"]: [badlisted["tag"]["value"]]},
                 )