diff --git a/microservices/trainer/trainer.py b/microservices/trainer/trainer.py
index ab9331b51fcb89b2200bbae7eb6c6de48029b870..03f572532f128b93a41067e4048132b540490ed2 100644
--- a/microservices/trainer/trainer.py
+++ b/microservices/trainer/trainer.py
@@ -18,7 +18,7 @@ from transformers.utils import (
 )
 
 import trainer_pb2_grpc
-from microservices.trainer import trainer_pb2
+import trainer_pb2
 
 is_busy = False