diff --git a/code-samples/community/serving/machinelearning-python-bentoml/iris_classifier.py b/code-samples/community/serving/machinelearning-python-bentoml/iris_classifier.py
index 324208bf47..dc880d5309 100644
--- a/code-samples/community/serving/machinelearning-python-bentoml/iris_classifier.py
+++ b/code-samples/community/serving/machinelearning-python-bentoml/iris_classifier.py
@@ -20,4 +20,4 @@ def predict(
             default=[[5.2, 2.3, 5.0, 0.7]]
         ),
     ) -> np.ndarray:
-        return self.artifacts.model.predict(df)
+        return self.model.predict(df)