diff --git a/web/src/engine/predictive-text/worker-thread/src/main/correction/context-tracker.ts b/web/src/engine/predictive-text/worker-thread/src/main/correction/context-tracker.ts
index 5a38b180ec..85babe47e4 100644
--- a/web/src/engine/predictive-text/worker-thread/src/main/correction/context-tracker.ts
+++ b/web/src/engine/predictive-text/worker-thread/src/main/correction/context-tracker.ts
@@ -315,7 +315,7 @@ export class ContextTracker {
               this.cache.add(transitionId, result);
             } else {
               this.cache.add(priorMatchState.transitionId, priorMatchState);
-              return this.cache.get(priorMatchState.transitionId);
+              return priorMatchState;
             }
           }