diff --git a/colab/demo.ipynb b/colab/demo.ipynb
index e92f8b04d..d1c58ad6c 100644
--- a/colab/demo.ipynb
+++ b/colab/demo.ipynb
@@ -292,9 +292,9 @@
         "if os.path.exists(f\"data/nerfstudio/{scene}/transforms.json\"):\n",
         "    !ns-train nerfacto --viewer.websocket-port 7007 nerfstudio-data --data data/nerfstudio/$scene --downscale-factor 4\n",
         "else:\n",
-        "    import IPython\n",
-        "    display(IPython.display.HTML('<h3 style=\"color:red\">Error: Data processing did not complete</h3>'))\n",
-        "    display(IPython.display.HTML('<h3>Please re-run `Downloading and Processing Data`, or view the FAQ for more info.</h3>'))"
+        "    from IPython.core.display import display, HTML\n",
+        "    display(HTML('<h3 style=\"color:red\">Error: Data processing did not complete</h3>'))\n",
+        "    display(HTML('<h3>Please re-run `Downloading and Processing Data`, or view the FAQ for more info.</h3>'))"
       ]
     },
     {
@@ -383,4 +383,4 @@
   },
   "nbformat": 4,
   "nbformat_minor": 0
-}
\ No newline at end of file
+}