diff --git a/pkg/common/db/unrelation/extend_msg.go b/pkg/common/db/unrelation/extend_msg.go
index 15ac95ca6..3eb902140 100644
--- a/pkg/common/db/unrelation/extend_msg.go
+++ b/pkg/common/db/unrelation/extend_msg.go
@@ -39,7 +39,7 @@ func (e *ExtendMsgSetMongoDriver) GetAllExtendMsgSet(ctx context.Context, ID str
 	if err != nil {
 		return nil, utils.Wrap(err, "")
 	}
-	err = cursor.All(context.Background(), &sets)
+	err = cursor.All(ctx, &sets)
 	if err != nil {
 		return nil, utils.Wrap(err, fmt.Sprintf("cursor is %s", cursor.Current.String()))
 	}