diff --git a/hamilton/plugins/h_pandera.py b/hamilton/plugins/h_pandera.py
index 7f0c8ddcc..1ff8f3e9c 100644
--- a/hamilton/plugins/h_pandera.py
+++ b/hamilton/plugins/h_pandera.py
@@ -51,7 +51,14 @@ def foo() -> DataFrame[MySchema]:
 
             from hamilton import function_modifiers
 
-            @function_modifiers.check_output(schema=MySchema)
+            schema = pa.DataFrameSchema({
+                "a": pa.Column(pa.Int),
+                "b": pa.Column(pa.Float),
+                "c": pa.Column(pa.String, nullable=True),
+                "d": pa.Column(pa.Float),
+            })
+
+            @function_modifiers.check_output(schema=schema)
             def foo() -> pd.DataFrame:
                 return pd.DataFrame() # will fail