diff --git a/packages/jupyter-ai/jupyter_ai/document_loaders/directory.py b/packages/jupyter-ai/jupyter_ai/document_loaders/directory.py
index b377cbdd7..b8f9de3bb 100644
--- a/packages/jupyter-ai/jupyter_ai/document_loaders/directory.py
+++ b/packages/jupyter-ai/jupyter_ai/document_loaders/directory.py
@@ -5,9 +5,9 @@
 from typing import List
 
 import dask
-from langchain.document_loaders import PyPDFLoader
 from langchain.schema import Document
 from langchain.text_splitter import TextSplitter
+from langchain_community.document_loaders import PyPDFLoader
 
 
 # Uses pypdf which is used by PyPDFLoader from langchain