diff --git a/src/modules/network_data/controller/upf.go b/src/modules/network_data/controller/upf.go
index aa3ef337..366fc15d 100644
--- a/src/modules/network_data/controller/upf.go
+++ b/src/modules/network_data/controller/upf.go
@@ -32,9 +32,8 @@ type UPFController struct {
 func (s *UPFController) TotalFlow(c *gin.Context) {
 	language := ctx.AcceptLanguage(c)
 	var querys struct {
-		NeType string `json:"neType" form:"neType" binding:"required"`
-		NeID   string `form:"neId" binding:"required"`
-		Day    int    `form:"day" binding:"required"`
+		NeID string `form:"neId" binding:"required"`
+		Day  int    `form:"day" binding:"required"`
 	}
 	if err := c.ShouldBindQuery(&querys); querys.Day < 0 || err != nil {
 		c.JSON(400, result.CodeMsg(400, i18n.TKey(language, "app.common.err400")))
@@ -42,7 +41,7 @@ func (s *UPFController) TotalFlow(c *gin.Context) {
 	}
 
 	// 查询网元获取IP
-	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID(querys.NeType, querys.NeID)
+	neInfo := s.neInfoService.SelectNeInfoByNeTypeAndNeID("UPF", querys.NeID)
 	if neInfo.NeId != querys.NeID || neInfo.IP == "" {
 		c.JSON(200, result.ErrMsg(i18n.TKey(language, "app.common.noNEInfo")))
 		return
diff --git a/src/modules/network_element/fetch_link/udm.go b/src/modules/network_element/fetch_link/udm.go
index 905bbb5b..d027f9f0 100644
--- a/src/modules/network_element/fetch_link/udm.go
+++ b/src/modules/network_element/fetch_link/udm.go
@@ -28,8 +28,7 @@ func UDMImportAuth(udmIP string, data map[string]any) (string, error) {
 	}
 
 	// 序列化结果
-	err = json.Unmarshal(resBytes, &resData)
-	if err != nil {
+	if err = json.Unmarshal(resBytes, &resData); err != nil {
 		logger.Errorf("UDMImportAuth Unmarshal %s", err.Error())
 		return "", err
 	}