diff --git a/src/message.go b/src/message.go
index 0d48d50..f2a14f0 100644
--- a/src/message.go
+++ b/src/message.go
@@ -93,7 +93,7 @@ type MsgJobReq struct {
 }
 
 type MsgJobReqPredict struct {
-	Code   int    `json:"code"`
-	Error  string `json:"error"`
-	Labels string `json:"labels"`
+	Code   int                `json:"code"`
+	Error  string             `json:"error"`
+	Labels map[string]float64 `json:"labels"`
 }
diff --git a/src/optimizer.go b/src/optimizer.go
index 0138571..5b39ef4 100644
--- a/src/optimizer.go
+++ b/src/optimizer.go
@@ -470,8 +470,7 @@ func (optimizer *Optimizer) PredictReq(job Job, role string) MsgJobReq {
 	var msg MsgJobReqPredict
 	err = json.Unmarshal([]byte(string(body)), &msg)
 	if err == nil && msg.Code == 0 {
-		var tmp map[string]float64
-		err = json.Unmarshal([]byte(msg.Labels), &tmp)
+		tmp := msg.Labels
 		if v, ok := tmp["cpu"]; ok {
 			req.CPU = int(math.Ceil(v / 100))
 		}