diff --git a/agenta-web/src/components/Playground/Views/TestView.tsx b/agenta-web/src/components/Playground/Views/TestView.tsx
index dded6b5f3..ae3305e6c 100644
--- a/agenta-web/src/components/Playground/Views/TestView.tsx
+++ b/agenta-web/src/components/Playground/Views/TestView.tsx
@@ -568,10 +568,10 @@ const App: React.FC<TestViewProps> = ({
             // String, FuncResponse or BaseResponse
             if (typeof result === "string") {
                 res = {version: "2.0", data: result} as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
             } else if (isFuncResponse(result)) {
                 res = {version: "2.0", data: result.message} as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
 
                 const {message, cost, latency, usage} = result
                 setAdditionalDataList((prev) => {
@@ -581,7 +581,7 @@ const App: React.FC<TestViewProps> = ({
                 })
             } else if (isBaseResponse(result)) {
                 res = result as BaseResponse
-                setResultForIndex(getStringOrJson(res), index)
+                setResultForIndex(getStringOrJson(res.data), index)
 
                 const {data, trace} = result
                 setAdditionalDataList((prev) => {
diff --git a/agenta-web/src/lib/helpers/utils.ts b/agenta-web/src/lib/helpers/utils.ts
index 49e2f3b93..aaef7b79d 100644
--- a/agenta-web/src/lib/helpers/utils.ts
+++ b/agenta-web/src/lib/helpers/utils.ts
@@ -338,11 +338,7 @@ export const getInitials = (str: string, limit = 2) => {
 }
 
 export const getStringOrJson = (value: any) => {
-    return typeof value === "string"
-        ? value
-        : typeof value?.data === "string"
-          ? value?.data
-          : JSON.stringify(value.data, null, 2)
+    return typeof value === "string" ? value : JSON.stringify(value, null, 2)
 }
 
 export const filterVariantParameters = ({