diff --git a/apps/web/src/lib/ai/gradeWorksheet.ts b/apps/web/src/lib/ai/gradeWorksheet.ts
index 88fadf80..9000d7c3 100644
--- a/apps/web/src/lib/ai/gradeWorksheet.ts
+++ b/apps/web/src/lib/ai/gradeWorksheet.ts
@@ -87,6 +87,7 @@ const gradingSchema = {
     'correctCount',
     'accuracy',
     'errorPatterns',
+    'currentStepEstimate',
     'suggestedStepId',
     'reasoning',
     'feedback',