diff --git a/src/zenml/models/v2/core/step_run.py b/src/zenml/models/v2/core/step_run.py
index c02433c412d..c018d13546b 100644
--- a/src/zenml/models/v2/core/step_run.py
+++ b/src/zenml/models/v2/core/step_run.py
@@ -46,6 +46,8 @@
 
 
 class StepRunInputResponse(ArtifactVersionResponse):
+    """Response model for step run inputs."""
+
     input_type: StepRunInputArtifactType