diff --git a/MLFlow Demo.ipynb b/MLFlow Demo.ipynb
index a67f9131a3f48738b655ad6928488698c381c5ac..110d46dc5508d36f7d39c7109a184841c1d24658 100644
--- a/MLFlow Demo.ipynb	
+++ b/MLFlow Demo.ipynb	
@@ -431,6 +431,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
+    "from datetime import datetime\n",
     "experiment_name = \"iris_classifier_basic\"\n",
     "run_name=\"iris_classifier_basic\" +str(datetime.now().strftime(\"%d-%m-%y\"))\n",
     "create_exp_and_register_model(experiment_name,run_name,run_metrics,model,'confusion_matrix.png')"