diff --git a/src/trainingdata/trainingdata.cc b/src/trainingdata/trainingdata.cc
index 1285dc7b49..b9d102e449 100644
--- a/src/trainingdata/trainingdata.cc
+++ b/src/trainingdata/trainingdata.cc
@@ -263,6 +263,8 @@ void V6TrainingDataArray::Add(const Node* node, const PositionHistory& history,
       DriftCorrect(result.root_q, result.root_d);
   std::tie(result.played_q, result.played_d) =
       DriftCorrect(result.played_q, result.played_d);
+  std::tie(result.orig_q, result.orig_d) =
+      DriftCorrect(result.orig_q, result.orig_d);
 
   result.root_m = node->GetM();
   result.best_m = best_eval.ml;