diff --git a/comm/pool.go b/comm/pool.go
index 3008726b2..48a44044d 100644
--- a/comm/pool.go
+++ b/comm/pool.go
@@ -1,13 +1,12 @@
 package comm
 
 import (
-	"go_dreamfactory/pb"
 	"sync"
 )
 
 var taskParamPool = &sync.Pool{
 	New: func() interface{} {
-		return &pb.AgentMessage{}
+		return &TaskParam{}
 	},
 }