diff --git a/haystack/retriever/sparse.py b/haystack/retriever/sparse.py
index 2062aace3..010cb3010 100644
--- a/haystack/retriever/sparse.py
+++ b/haystack/retriever/sparse.py
@@ -186,7 +186,7 @@ class TfidfRetriever(BaseRetriever):
         for para, meta in zip(paragraphs, meta_data):
             documents.append(
                 Document(
-                    id=meta["paragraph_id"],
+                    id=meta["document_id"],
                     text=para,
                     meta=meta.get("meta", {})
                 ))
diff --git a/test/test_finder.py b/test/test_finder.py
index 7bbfde5dd..82ae7447e 100644
--- a/test/test_finder.py
+++ b/test/test_finder.py
@@ -16,7 +16,6 @@ def test_finder_get_answers(reader, document_store_with_docs):
     assert prediction["answers"][0]["probability"] >= 0
     assert prediction["answers"][0]["meta"]["meta_field"] == "test1"
     assert prediction["answers"][0]["context"] == "My name is Carla and I live in Berlin"
-    assert prediction["answers"][0]["document_id"] == "0"
 
     assert len(prediction["answers"]) == 3
 
diff --git a/test/test_tfidf_retriever.py b/test/test_tfidf_retriever.py
index fffcdde03..42677fac8 100644
--- a/test/test_tfidf_retriever.py
+++ b/test/test_tfidf_retriever.py
@@ -18,7 +18,7 @@ def test_tfidf_retriever():
     retriever.fit()
     assert retriever.retrieve("godzilla", top_k=1) == [
         Document(
-            id='0',
+            id='26f84672c6d7aaeb8e2cd53e9c62d62d',
             text='godzilla says hello',
             external_source_id=None,
             question=None,