diff --git a/agent-node/protos/agent_pb2_grpc.py b/agent-node/protos/agent_pb2_grpc.py
index 932d45e..b91c8a0 100644
--- a/agent-node/protos/agent_pb2_grpc.py
+++ b/agent-node/protos/agent_pb2_grpc.py
@@ -2,7 +2,7 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
-import agent_pb2 as agent__pb2
+from . import agent_pb2 as agent__pb2
 
 
 class AgentOrchestratorStub(object):
diff --git a/ai-hub/app/protos/agent_pb2_grpc.py b/ai-hub/app/protos/agent_pb2_grpc.py
index 932d45e..b91c8a0 100644
--- a/ai-hub/app/protos/agent_pb2_grpc.py
+++ b/ai-hub/app/protos/agent_pb2_grpc.py
@@ -2,7 +2,7 @@
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
 
-import agent_pb2 as agent__pb2
+from . import agent_pb2 as agent__pb2
 
 
 class AgentOrchestratorStub(object):