diff --git a/pkg/common/db/batch_insert_chat.go b/pkg/common/db/batch_insert_chat.go
index 7ed0d7e1a..54444a46f 100644
--- a/pkg/common/db/batch_insert_chat.go
+++ b/pkg/common/db/batch_insert_chat.go
@@ -3,7 +3,7 @@ package db
 import (
 	"Open_IM/pkg/common/config"
 	"Open_IM/pkg/common/log"
-	pbMsg "Open_IM/pkg/proto/chat"
+	pbMsg "Open_IM/pkg/proto/msg"
 	"Open_IM/pkg/utils"
 	"context"
 	"errors"
diff --git a/pkg/common/db/redisModel_test.go b/pkg/common/db/redisModel_test.go
index 70c7f4d22..66025f6ab 100644
--- a/pkg/common/db/redisModel_test.go
+++ b/pkg/common/db/redisModel_test.go
@@ -2,7 +2,7 @@ package db
 
 import (
 	"Open_IM/pkg/common/constant"
-	pbChat "Open_IM/pkg/proto/chat"
+	pbChat "Open_IM/pkg/proto/msg"
 	server_api_params "Open_IM/pkg/proto/sdk_ws"
 	"context"
 	"flag"