diff --git a/runtime/datamate-python/app/module/shared/util/structured_file.py b/runtime/datamate-python/app/module/shared/util/structured_file.py
index 040c2ba5..ef3232c6 100644
--- a/runtime/datamate-python/app/module/shared/util/structured_file.py
+++ b/runtime/datamate-python/app/module/shared/util/structured_file.py
@@ -73,11 +73,11 @@ def __init__(self):
         self.schema = {
             "type": "object",
             "properties": {
-                "question": {"type": "string"},
-                "conclusion": {"type": "string"},
+                "instruction": {"type": "string"},
+                "output": {"type": "string"},
                 "chain_of_thought": {"type": "string"}
             },
-            "required": ["question", "conclusion", "chain_of_thought"],
+            "required": ["question", "instruction", "output"],
         }
         self.schema_list = {
             "type": "array",