diff --git a/Zadanie_09_MLflow_Predict_Registry.py b/Zadanie_09_MLflow_Predict_Registry.py
index 526ae99..03f86a8 100644
--- a/Zadanie_09_MLflow_Predict_Registry.py
+++ b/Zadanie_09_MLflow_Predict_Registry.py
@@ -10,7 +10,7 @@ for mv in client.search_model_versions(f"name='{model_name}'"):
     if int(mv.version) > version:
         version = int(mv.version)
 
-model = mlflow.pytorch.load_model(
+model = mlflow.keras.load_model(
     model_uri=f"models:/{model_name}/{version}"
 )