From bc160812a1c5b9586fd621185fa11a5db81153dd Mon Sep 17 00:00:00 2001
From: Miodec <bartnikjack@gmail.com>
Date: Thu, 20 Jan 2022 23:23:31 +0100
Subject: [PATCH] fixed burst heatmap not working in zen. closes #2327

---
 src/js/test/test-stats.js | 3 +++
 src/js/test/test-ui.js    | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/js/test/test-stats.js b/src/js/test/test-stats.js
index 5df5f9cb7..44223f27a 100644
--- a/src/js/test/test-stats.js
+++ b/src/js/test/test-stats.js
@@ -237,6 +237,9 @@ export function calculateBurst() {
   let wordLength;
   if (Config.mode === "zen") {
     wordLength = TestLogic.input.current.length;
+    if (wordLength == 0) {
+      wordLength = TestLogic.input.getHistoryLast().length;
+    }
   } else {
     wordLength = TestLogic.words.getCurrent().length;
   }
diff --git a/src/js/test/test-ui.js b/src/js/test/test-ui.js
index caa95fe97..518cc2d7c 100644
--- a/src/js/test/test-ui.js
+++ b/src/js/test/test-ui.js
@@ -907,7 +907,7 @@ export function applyBurstHeatmap() {
 
     if (
       TestLogic.input.getHistory(TestLogic.input.getHistory().length - 1)
-        .length !== TestLogic.words.getCurrent().length
+        .length !== TestLogic.words.getCurrent()?.length
     ) {
       burstlist = burstlist.splice(0, burstlist.length - 1);
     }