diff --git a/notebooks/llamaindex-load.ipynb b/notebooks/llamaindex-load.ipynb
index 7d965e1..c598f63 100644
--- a/notebooks/llamaindex-load.ipynb
+++ b/notebooks/llamaindex-load.ipynb
@@ -94,7 +94,7 @@
    "outputs": [],
    "source": [
     "# Load query engine from local folder, 'index'\n",
-    "from llama_index import StorageContext, load_index_from_storage\n",
+    "from llama_index.core import StorageContext, load_index_from_storage\n",
     "\n",
     "# rebuild storage context\n",
     "storage_context = StorageContext.from_defaults(persist_dir=\"./index\")\n",