diff --git a/packages/hydrooj/src/handler/training.ts b/packages/hydrooj/src/handler/training.ts
index c6553c0a..f950ebd1 100644
--- a/packages/hydrooj/src/handler/training.ts
+++ b/packages/hydrooj/src/handler/training.ts
@@ -149,7 +149,7 @@ class TrainingDetailHandler extends Handler {
             if (nsdoc.isDone) doneNids.add(node._id);
             nsdict[node._id] = nsdoc;
         }
-        const tsdoc = await training.setStatus(domainId, tdoc.docId, this.user._id, {
+        const tsdoc = await training.setStatus(domainId, tdoc.docId, udoc._id, {
             doneNids: Array.from(doneNids),
             donePids: Array.from(donePids),
             done: doneNids.size === tdoc.dag.length,