.

CERT-Polska · Dec 20, 2024 · a6b8916 · a6b8916
1 parent d80f4dd
commit a6b8916
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/artemis/modules/ip_lookup.py b/artemis/modules/ip_lookup.py
@@ -20,12 +20,12 @@ class IPLookup(ArtemisBase):
         {"type": TaskType.DOMAIN.value},
     ]
 
-    def _process(self, current_task: Task, domain: str) -> Set[str]:
+    def _process(self, current_task: Task, domain: str) -> List[str]:
         found_ips = lookup(domain)
         for found_ip in found_ips:
             new_task = Task({"type": TaskType.NEW}, payload={"data": found_ip})
             self.add_task(current_task, new_task)
-        return found_ips
+        return list(found_ips)
 
     def run(self, current_task: Task) -> None:
         domain = current_task.get_payload(TaskType.DOMAIN)