diff --git a/Makefile b/Makefile
index 6ea08f4..8057f5a 100755
--- a/Makefile
+++ b/Makefile
@@ -49,6 +49,7 @@ PROTOS = \
 	v1/vald/filter.proto \
 	v1/vald/insert.proto \
 	v1/vald/object.proto \
+	v1/vald/index.proto \
 	v1/vald/remove.proto \
 	v1/vald/search.proto \
 	v1/vald/update.proto \
diff --git a/src/tests/test_e2e.py b/src/tests/test_e2e.py
index f73e659..bc44ca8 100644
--- a/src/tests/test_e2e.py
+++ b/src/tests/test_e2e.py
@@ -10,6 +10,7 @@
 from vald.v1.vald import upsert_pb2_grpc
 from vald.v1.vald import remove_pb2_grpc
 from vald.v1.vald import object_pb2_grpc
+from vald.v1.vald import index_pb2_grpc
 from vald.v1.payload import payload_pb2
 
 
@@ -79,7 +80,7 @@ def test_save_index(self):
         self.assertIsInstance(result, payload_pb2.Empty)
 
     def test_index_info(self):
-        stub = agent_pb2_grpc.AgentStub(self.channel)
+        stub = index_pb2_grpc.IndexStub(self.channel)
         result = stub.IndexInfo(payload_pb2.Empty())
         self.assertIsInstance(result, payload_pb2.Info.Index.Count)
         self.assertEqual(result.stored, 99)