From bdaa4c51c980d8b6081ba7ba9899449b2e918c46 Mon Sep 17 00:00:00 2001
From: terrtia <or1994@hotmail.fr>
Date: Tue, 9 Jan 2024 12:15:40 +0100
Subject: [PATCH] fix: [hosts] fix number of hosts extracted

---
 bin/modules/DomClassifier.py | 1 -
 bin/modules/Hosts.py         | 2 +-
 2 files changed, 1 insertion(+), 2 deletions(-)

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 @@ sys.path.append(os.environ['AIL_BIN'])
 # 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 @@ class Hosts(AbstractModule):
         # 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: