fix(words generator): infinite custom text tests not working correctly

monkeytypegame · May 8, 2024 · 95f5e6c · 95f5e6c
1 parent 39dd905
commit 95f5e6c
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/frontend/src/ts/test/words-generator.ts b/frontend/src/ts/test/words-generator.ts
@@ -417,11 +417,7 @@ export function getWordsLimit(): number {
 
   //custom
   if (Config.mode === "custom") {
-    if (
-      CustomText.getLimitValue() === 0 ||
-      CustomText.getLimitMode() === "time" ||
-      CustomText.getLimitMode() === "section"
-    ) {
+    if (CustomText.getLimitValue() === 0) {
       limit = 100;
     } else {
       limit =
@@ -446,7 +442,8 @@ export function getWordsLimit(): number {
   if (
     Config.mode === "custom" &&
     CustomText.getLimitMode() === "word" &&
-    CustomText.getLimitValue() < limit
+    CustomText.getLimitValue() < limit &&
+    CustomText.getLimitValue() !== 0
   ) {
     limit = CustomText.getLimitValue();
   }