diff --git a/api/handler/chat_handler.go b/api/handler/chat_handler.go
index 0fae34f4..baaa1e86 100644
--- a/api/handler/chat_handler.go
+++ b/api/handler/chat_handler.go
@@ -93,7 +93,7 @@ func (h *ChatHandler) ChatHandle(c *gin.Context) {
 		Id:       chatModel.Id,
 		Value:    chatModel.Value,
 		Platform: types.Platform(chatModel.Platform)}
-	logger.Infof("New websocket connected, IP: %s, Username: %s", c.Request.RemoteAddr, session.Username)
+	logger.Infof("New websocket connected, IP: %s, Username: %s", c.ClientIP(), session.Username)
 	var chatRole model.ChatRole
 	res = h.db.First(&chatRole, roleId)
 	if res.Error != nil || !chatRole.Enable {
diff --git a/api/handler/mj_handler.go b/api/handler/mj_handler.go
index 82550be3..1c446952 100644
--- a/api/handler/mj_handler.go
+++ b/api/handler/mj_handler.go
@@ -85,7 +85,7 @@ func (h *MidJourneyHandler) Notify(c *gin.Context) {
 	logger.Debugf("收到 MidJourney 回调请求：%+v", data)
 	h.lock.Lock()
 	defer h.lock.Unlock()
-	
+
 	// the job is saved
 	var job model.MidJourneyJob
 	res := h.db.Where("message_id = ?", data.MessageId).First(&job)