diff --git a/embedchain/app.py b/embedchain/app.py
index 17d6da1b..0906a3c6 100644
--- a/embedchain/app.py
+++ b/embedchain/app.py
@@ -254,13 +254,7 @@ class App(EmbedChain):
             )
             result = []
             for c in context:
-                result.append(
-                    {
-                        "context": c[0],
-                        "source": c[1],
-                        "document_id": c[2],
-                    }
-                )
+                result.append({"context": c[0], "metadata": c[1]})
             return result
         else:
             # Make API call to the backend to get the results
diff --git a/pyproject.toml b/pyproject.toml
index 4e857fab..650130e8 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "embedchain"
-version = "0.1.49"
+version = "0.1.50"
 description = "Data platform for LLMs - Load, index, retrieve and sync any unstructured data"
 authors = [
     "Taranjeet Singh <taranjeet@embedchain.ai>",