diff --git a/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java b/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java
index 9d28380b..45806397 100644
--- a/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java
+++ b/protocol/src/main/java/com/zfoo/protocol/serializer/protobuf/GeneratePbUtils.java
@@ -252,6 +252,10 @@ public abstract class GeneratePbUtils {
         var imports = new HashSet<String>();
         imports.add(Protocol.class.getName());
 
+        if (CollectionUtils.isNotEmpty(pbMessage.getComments())) {
+            imports.add(Note.class.getName());
+        }
+
         var pbFields = pbMessage.getFields();
         for (var pbField : pbFields) {
             if (CollectionUtils.isNotEmpty(pbField.getComments())) {