diff --git a/notebooks/classification.ipynb b/notebooks/classification.ipynb
index ae486b8..c613909 100644
--- a/notebooks/classification.ipynb
+++ b/notebooks/classification.ipynb
@@ -8,8 +8,8 @@
    "source": [
     "import json\n",
     "\n",
-    "from promptify.models.nlp.openai_model import OpenAI\n",
-    "from promptify.prompts.nlp.prompter import Prompter"
+    "from promptify import OpenAI\n",
+    "from promptify import Prompter"
    ]
   },
   {