diff --git a/clarifai/client/model.py b/clarifai/client/model.py
index 81674396..af14d969 100644
--- a/clarifai/client/model.py
+++ b/clarifai/client/model.py
@@ -1233,8 +1233,9 @@ def _get_model_info_for_inference(self, inference_params: Dict = {},
 
     model_info = resources_pb2.Model()
     model_info.CopyFrom(self.model_info)
-    model_info.model_version.output_info.params.update(inference_params)
-    model_info.model_version.output_info.output_config.update(output_config)
+    model_info.model_version.output_info.params = inference_params
+    model_info.model_version.output_info.output_config.CopyFrom(
+        resources_pb2.OutputConfig(**output_config))
     return model_info
 
   def _list_concepts(self) -> List[str]: