diff --git a/notebooks/astradb_haystack_integration.ipynb b/notebooks/astradb_haystack_integration.ipynb
index e811b60..3db3e35 100644
--- a/notebooks/astradb_haystack_integration.ipynb
+++ b/notebooks/astradb_haystack_integration.ipynb
@@ -345,7 +345,7 @@
         "from haystack.components.builders.answer_builder import AnswerBuilder\n",
         "from haystack.components.builders.prompt_builder import PromptBuilder\n",
         "from haystack.components.generators import OpenAIGenerator\n",
-        "from haystack_integrations.components.retrievers.astra import AstraRetriever\n",
+        "from haystack_integrations.components.retrievers.astra import AstraEmbeddingRetriever\n",
         "\n",
         "prompt_template = \"\"\"\n",
         "                Given these documents, answer the question.\n",
@@ -362,7 +362,7 @@
         "    instance=SentenceTransformersTextEmbedder(model=embedding_model_name),\n",
         "    name=\"embedder\",\n",
         ")\n",
-        "rag_pipeline.add_component(instance=AstraRetriever(document_store=document_store), name=\"retriever\")\n",
+        "rag_pipeline.add_component(instance=AstraEmbeddingRetriever(document_store=document_store), name=\"retriever\")\n",
         "rag_pipeline.add_component(instance=PromptBuilder(template=prompt_template), name=\"prompt_builder\")\n",
         "rag_pipeline.add_component(instance=OpenAIGenerator(), name=\"llm\")\n",
         "rag_pipeline.add_component(instance=AnswerBuilder(), name=\"answer_builder\")\n",