diff --git a/reproduce/batch_eval.py b/reproduce/batch_eval.py
index a85e1ede..424b4f54 100644
--- a/reproduce/batch_eval.py
+++ b/reproduce/batch_eval.py
@@ -57,6 +57,10 @@ def batch_eval(query_file, result1_file, result2_file, output_file_path):
                 "Winner": "[Answer 1 or Answer 2]",
                 "Explanation": "[Provide explanation here]"
             }},
+            "Diversity": {{
+                "Winner": "[Answer 1 or Answer 2]",
+                "Explanation": "[Provide explanation here]"
+            }},
             "Empowerment": {{
                 "Winner": "[Answer 1 or Answer 2]",
                 "Explanation": "[Provide explanation here]"