diff --git a/docs/docs/integrations/tools/apify.ipynb b/docs/docs/integrations/tools/apify.ipynb
index 949bc93efad..96e2fef7a99 100644
--- a/docs/docs/integrations/tools/apify.ipynb
+++ b/docs/docs/integrations/tools/apify.ipynb
@@ -41,8 +41,8 @@
    "outputs": [],
    "source": [
     "from langchain.indexes import VectorstoreIndexCreator\n",
-    "from langchain_community.document_loaders.base import Document\n",
-    "from langchain_community.utilities import ApifyWrapper"
+    "from langchain_community.utilities import ApifyWrapper\n",
+    "from langchain_core.documents import Document"
    ]
   },
   {