diff --git a/agent/component/retrieval.py b/agent/component/retrieval.py
index a6f4048d9..6a3f89b74 100644
--- a/agent/component/retrieval.py
+++ b/agent/component/retrieval.py
@@ -108,6 +108,6 @@ class Retrieval(ComponentBase, ABC):
         df["content"] = df["content_with_weight"]
         del df["content_with_weight"]
         logging.debug("{} {}".format(query, df))
-        return df
+        return df.dropna()