diff --git a/docs/extras/integrations/document_loaders/assemblyai.ipynb b/docs/extras/integrations/document_loaders/assemblyai.ipynb
index 33fdef929d8..80d60df48b5 100644
--- a/docs/extras/integrations/document_loaders/assemblyai.ipynb
+++ b/docs/extras/integrations/document_loaders/assemblyai.ipynb
@@ -53,7 +53,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.assemblyai import AssemblyAIAudioTranscriptLoader\n",
+    "from langchain.document_loaders import AssemblyAIAudioTranscriptLoader\n",
     "\n",
     "audio_file = \"https://storage.googleapis.com/aai-docs-samples/nbc.mp3\"\n",
     "# or a local file path: audio_file = \"./nbc.mp3\"\n",
@@ -148,10 +148,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from langchain.document_loaders.assemblyai import (\n",
-    "    AssemblyAIAudioTranscriptLoader,\n",
-    "    TranscriptFormat,\n",
-    ")\n",
+    "from langchain.document_loaders.assemblyai import TranscriptFormat\n",
     "\n",
     "loader = AssemblyAIAudioTranscriptLoader(\n",
     "    file_path=\"./your_file.mp3\",\n",
diff --git a/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py b/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
index a9b6112e7b1..550a2c0d1aa 100644
--- a/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
+++ b/libs/langchain/tests/unit_tests/document_loaders/test_assemblyai.py
@@ -1,10 +1,8 @@
 import pytest
 from pytest_mock import MockerFixture
 
-from langchain.document_loaders.assemblyai import (
-    AssemblyAIAudioTranscriptLoader,
-    TranscriptFormat,
-)
+from langchain.document_loaders import AssemblyAIAudioTranscriptLoader
+from langchain.document_loaders.assemblyai import TranscriptFormat
 
 
 @pytest.mark.requires("assemblyai")