diff --git a/protoc_plugins/schema2dict.py b/protoc_plugins/schema2dict.py
index 3a7a63b..3456edc 100644
--- a/protoc_plugins/schema2dict.py
+++ b/protoc_plugins/schema2dict.py
@@ -21,7 +21,8 @@
 """
 import json
 import sys
-from chameleon.protos import schema_pb2, annotations_pb2
+from chameleon.protos import schema_pb2
+from google.api import annotations_pb2
 from protobuf_to_dict import protobuf_to_dict