diff --git a/repo/main.py b/repo/main.py
index 4473248..dd11c6e 100644
--- a/repo/main.py
+++ b/repo/main.py
@@ -43,7 +43,7 @@ Score the draft on the following criteria (0–10):
 - Type hints usage
 - Edge case handling
 - Naming conventions
-Return a JSON object with keys: pep8, type_hints, edge_cases, naming. Also provide the weakest criterion and verdict ("ok" if all scores >=7 else "needs_revision").'''
+Return a JSON object with keys: pep8, type_hints, edge_cases, naming, weakest_criterion, verdict. The verdict should be "ok" if all scores are >=7, otherwise "needs_revision".'''
     response = await llm.invoke(prompt)
     import json
     try: