diff --git a/integrations/ollama/tests/test_document_embedder.py b/integrations/ollama/tests/test_document_embedder.py
index 55514feed..a5694db33 100644
--- a/integrations/ollama/tests/test_document_embedder.py
+++ b/integrations/ollama/tests/test_document_embedder.py
@@ -47,5 +47,5 @@ def test_run(self):
         reply = embedder.run(list_of_docs)
 
         assert isinstance(reply, dict)
-        assert all(isinstance(element, float) for element in reply["embedding"])
+        assert all(isinstance(element, float) for element in reply["documents"][0].embedding)
         assert reply["meta"]["model"] == "orca-mini"