diff --git a/bin/modules/DomClassifier.py b/bin/modules/DomClassifier.py index 505ef53c..94cf53db 100755 --- a/bin/modules/DomClassifier.py +++ b/bin/modules/DomClassifier.py @@ -22,7 +22,6 @@ # Import Project packages ################################## from modules.abstract_module import AbstractModule -from lib.objects.Items import Item from lib.ConfigLoader import ConfigLoader from lib import d4 diff --git a/bin/modules/Hosts.py b/bin/modules/Hosts.py index fce5595e..979e6680 100755 --- a/bin/modules/Hosts.py +++ b/bin/modules/Hosts.py @@ -55,7 +55,7 @@ def compute(self, message): # if mimetype.split('/')[0] == "text": content = item.get_content() - hosts = self.regex_findall(self.host_regex, item.get_id(), content) + hosts = self.regex_findall(self.host_regex, item.get_id(), content, r_set=True) if hosts: print(f'{len(hosts)} host {item.get_id()}') for host in hosts: