Small fixes to previous

ouspg · May 29, 2024 · 561372e · 561372e
1 parent 603a3c4
commit 561372e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tcsfw/builder_backend.py b/tcsfw/builder_backend.py
@@ -230,6 +230,7 @@ def external_activity(self, value: ExternalActivity) -> Self:
 
     def in_networks(self, *network: NetworkBuilder) -> Self:
         self.entity.networks = [n.network for n in network]
+        return self
 
     def software(self, name: Optional[str] = None) -> 'SoftwareBackend':
         if name is None:

diff --git a/tcsfw/traffic.py b/tcsfw/traffic.py
@@ -111,7 +111,7 @@ def __eq__(self, v) -> bool:
 
 class ServiceScan(Event):
     """Individual service scan result"""
-    def __init__(self, evidence: Evidence, endpoint: EndpointAddress, service_name=""):
+    def __init__(self, evidence: Evidence, endpoint: AnyAddress, service_name=""):
         super().__init__(evidence)
         self.endpoint = endpoint
         self.service_name = service_name