diff --git a/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py b/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py
index 70d4f656d..3913f9e69 100644
--- a/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py
+++ b/examples/pandera_plugin/pandera_plugin/validating_and_testing_ml_pipelines.py
@@ -118,7 +118,7 @@
 
 
 # %%
-class RawData(pa.SchemaModel):
+class RawData(pa.DataFrameModel):
     age: Series[int] = pa.Field(in_range={"min_value": 0, "max_value": 200})
     sex: Series[int] = pa.Field(isin=[0, 1])
     cp: Series[int] = pa.Field(
diff --git a/examples/pandera_plugin/requirements.in b/examples/pandera_plugin/requirements.in
index 6cc17e15b..f3ce15966 100644
--- a/examples/pandera_plugin/requirements.in
+++ b/examples/pandera_plugin/requirements.in
@@ -6,5 +6,5 @@ flytekitplugins-pandera>=0.16.0
 hypothesis
 joblib
 pandas
-pandera>=0.7.0
+pandera>=0.20.0
 scikit-learn