diff --git a/notebooks/metadata_enrichment.ipynb b/notebooks/metadata_enrichment.ipynb
index 844213b..6ef3dd0 100644
--- a/notebooks/metadata_enrichment.ipynb
+++ b/notebooks/metadata_enrichment.ipynb
@@ -52,7 +52,7 @@
       },
       "outputs": [],
       "source": [
-        "from haystack import Document, component\n",
+        "from haystack import Document, component, Pipeline\n",
         "from haystack.components.builders import PromptBuilder\n",
         "from haystack.components.converters import HTMLToDocument\n",
         "from haystack.components.fetchers import LinkContentFetcher\n",
@@ -60,9 +60,8 @@
         "from haystack.components.generators.openai_utils import _convert_message_to_openai_format\n",
         "from haystack.components.preprocessors import DocumentSplitter\n",
         "from haystack.dataclasses import ChatMessage, StreamingChunk\n",
-        "from haystack.document_stores.in_memory import InMemoryDocumentStore\n",
         "\n",
-        "from openai import OpenAI, Stream\n",
+        "from openai import Stream\n",
         "from openai.types.chat import ChatCompletion, ChatCompletionChunk\n",
         "from typing import List, Any, Dict, Optional, Callable, Union\n",
         "from pydantic import BaseModel"
@@ -182,9 +181,6 @@
       },
       "outputs": [],
       "source": [
-        "from haystack import component, Pipeline, Document\n",
-        "from haystack.components.builders.prompt_builder import PromptBuilder\n",
-        "\n",
         "DEFAULT_PROMPT = \"\"\"\n",
         "Given the contents of the documents, extract the requested metadata.\n",
         "The requested metadata is {{ metadata_model }}\n",