From 6920b3350d36d1d2e51fabdab70cf61d59554cd3 Mon Sep 17 00:00:00 2001
From: Joshua Horton <joshua_horton@sil.org>
Date: Fri, 23 Jan 2026 09:03:38 -0600
Subject: [PATCH] change(web): address review comments

---
 .../src/main/correction/distance-modeler.ts            | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/web/src/engine/predictive-text/worker-thread/src/main/correction/distance-modeler.ts b/web/src/engine/predictive-text/worker-thread/src/main/correction/distance-modeler.ts
index 1b0c9ab633..31e62eff38 100644
--- a/web/src/engine/predictive-text/worker-thread/src/main/correction/distance-modeler.ts
+++ b/web/src/engine/predictive-text/worker-thread/src/main/correction/distance-modeler.ts
@@ -657,12 +657,14 @@ export async function *getBestMatches(searchModules: SearchQuotientNode[], timer
   do {
     const entry: SearchResult = timer.time(() => {
       if((priorResultsQueue.peek()?.totalCost ?? Number.POSITIVE_INFINITY) < spaceQueue.peek().currentCost) {
-        return priorResultsQueue.dequeue();
+        const result = priorResultsQueue.dequeue();
+        currentReturns[result.node.resultKey] = result.node;
+        return result;
       }
 
-      let bestQueue = spaceQueue.dequeue();
-      let newResult: PathResult = bestQueue.handleNextNode();
-      spaceQueue.enqueue(bestQueue);
+      let lowestCostSource = spaceQueue.dequeue();
+      let newResult: PathResult = lowestCostSource.handleNextNode();
+      spaceQueue.enqueue(lowestCostSource);
 
       if(newResult.type == 'none') {
         return null;