diff --git a/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py b/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
index 5b5372240..0fc883bec 100644
--- a/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
+++ b/llama_stack/providers/impls/meta_reference/inference/parallel_utils.py
@@ -124,7 +124,7 @@ def mp_rank_0() -> bool:
 
 
 def encode_msg(msg: ProcessingMessage) -> bytes:
-    return msg.model_dump_json().encode("utf-8")
+    return ProcessingMessageWrapper(payload=msg).model_dump_json().encode("utf-8")
 
 
 def retrieve_requests(reply_socket_url: str):