diff --git a/repository/test.data/warning.go b/repository/test.data/warning.go
index 8fb9c2d..95af60d 100644
--- a/repository/test.data/warning.go
+++ b/repository/test.data/warning.go
@@ -17,7 +17,8 @@ func UpdateWarning(req *model.Warning) error {
 			ReturnProbabilityLimitH:    req.ReturnProbabilityLimitH,
 			PassProbabilityLimitL:      req.PassProbabilityLimitL,
 			AveragePassProbability:     req.AveragePassProbability,
-		}).Find(&warning)
+		})
+		global.PostGreSQL.Where("product = ? AND step = ?", req.Product, req.Step).Find(&warning)
 	} else {
 		global.PostGreSQL.Model(&warning).Updates(map[string]interface{}{
 			"first_pass_probability_limit_l": req.FirstPassProbabilityLimitL,