diff --git a/demo/Diffusion/models.py b/demo/Diffusion/models.py
index d9d82b69..06929dad 100644
--- a/demo/Diffusion/models.py
+++ b/demo/Diffusion/models.py
@@ -596,7 +596,7 @@ def optimize(self, onnx_graph):
         opt.infer_shapes()
         opt.info(self.name + ': shape inference')
         opt.select_outputs([0], names=['text_embeddings']) # rename network output
-        opt.info(self.name + ': remove output[0]')
+        opt.info(self.name + ': rename output[0]')
         opt_onnx_graph = opt.cleanup(return_onnx=True)
         if 'hidden_states' in self.extra_output_names:
             opt_onnx_graph = opt.clip_add_hidden_states(return_onnx=True)