diff --git a/src/diffusers/modular_pipelines/modular_pipeline_utils.py b/src/diffusers/modular_pipelines/modular_pipeline_utils.py
index 5468cf54d0..8116f26d39 100644
--- a/src/diffusers/modular_pipelines/modular_pipeline_utils.py
+++ b/src/diffusers/modular_pipelines/modular_pipeline_utils.py
@@ -549,7 +549,7 @@ class InputParam:
 class OutputParam:
     """Specification for an output parameter."""
 
-    name: str = None
+    name: str 
     type_hint: Any = None
     description: str = ""
     kwargs_type: str = None
diff --git a/src/diffusers/modular_pipelines/qwenimage/before_denoise.py b/src/diffusers/modular_pipelines/qwenimage/before_denoise.py
index 418d927f4f..aae6eb50d9 100644
--- a/src/diffusers/modular_pipelines/qwenimage/before_denoise.py
+++ b/src/diffusers/modular_pipelines/qwenimage/before_denoise.py
@@ -674,7 +674,7 @@ class QwenImageLayeredSetTimestepsStep(ModularPipelineBlocks):
         return [
             InputParam.template("num_inference_steps"),
             InputParam.template("sigmas"),
-            InputParam.template("image_latents", note="Can be generated from vae encoder and packed in input step."),
+            InputParam.template("image_latents"),
         ]
 
     @property