diff --git a/Model Validation.ipynb b/Model Validation.ipynb
index 65e9f0e..68eb54d 100644
--- a/Model Validation.ipynb	
+++ b/Model Validation.ipynb	
@@ -289,7 +289,8 @@
     "validation_data_df = spark.table(f\"{catalog}.{db}.{validation_data}\").toPandas()\n",
     "\n",
     "requirements = mlflow.pyfunc.get_model_dependencies(model_uri=f\"models:/{catalog}.{db}.{model_name}@Challenger\")\n",
-    "%pip install -r {requirements}"
+    "%pip install -r {requirements}\n",
+    "%pip install --force-reinstall databricks-automl-runtime==0.2.20.6"
    ]
   },
   {