diff --git a/src/l3_attackmitigator/service/l3_attackmitigatorService.py b/src/l3_attackmitigator/service/l3_attackmitigatorService.py
index a359d1d9fc897ea3c01c7fa4d39e9dfbb4195306..2ce7f93169430d586ba33a03f03918966c8dcc7b 100644
--- a/src/l3_attackmitigator/service/l3_attackmitigatorService.py
+++ b/src/l3_attackmitigator/service/l3_attackmitigatorService.py
@@ -21,7 +21,7 @@ from grpc_health.v1.health_pb2_grpc import add_HealthServicer_to_server
 from common.proto.l3_attackmitigator_pb2_grpc import (
     add_L3AttackmitigatorServicer_to_server,
 )
-from common.service.l3_attackmitigatorServiceServicerImpl import (
+from l3_attackmitigator.service.l3_attackmitigatorServiceServicerImpl import (
     l3_attackmitigatorServiceServicerImpl,
 )
 from l3_attackmitigator.Config import (
diff --git a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
index ead32d248bf4f22ecab0ce9de4ae364aa4925995..ad05b0ee62e87ce9028dc043b693c1b4cae008b3 100644
--- a/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
+++ b/src/l3_centralizedattackdetector/service/l3_centralizedattackdetectorServiceServicerImpl.py
@@ -27,7 +27,7 @@ from common.proto.l3_centralizedattackdetector_pb2_grpc import (
 )
 
 from common.proto.l3_attackmitigator_pb2 import (
-    Output,
+    L3AttackmitigatorOutput,
 )
 from common.proto.l3_attackmitigator_pb2_grpc import (
     L3AttackmitigatorStub,
@@ -88,7 +88,7 @@ class l3_centralizedattackdetectorServiceServicerImpl(L3Centralizedattackdetecto
             output_message["tag_name"] = "Normal"
             output_message["tag"] = 0
 
-        return Output(**output_message)
+        return L3AttackmitigatorOutput(**output_message)
 
     def SendInput(self, request, context):
         # PERFORM INFERENCE WITH SENT INPUTS