From 65eff470cfe7e52ec3c778290dc0c20d046ee23a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABl=20van=20Noord?=
 <13665637+DanielNoord@users.noreply.github.com>
Date: Sun, 25 Jul 2021 14:41:51 +0200
Subject: [PATCH] Fix bug with storing locked buildings

---
 src/Disp/buildingTiles/toggleBuildingLock.js | 20 ++++++++++++++++----
 1 file changed, 16 insertions(+), 4 deletions(-)

diff --git a/src/Disp/buildingTiles/toggleBuildingLock.js b/src/Disp/buildingTiles/toggleBuildingLock.js
index 0fa2b5d..3b84a27 100644
--- a/src/Disp/buildingTiles/toggleBuildingLock.js
+++ b/src/Disp/buildingTiles/toggleBuildingLock.js
@@ -4,18 +4,30 @@
  */
 export default function toggleBuildingLock(index) {
   if (l(`productLock${index}`).innerHTML === 'Lock') {
+    // Add to storing array
     Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames.push(
       index.toString(),
     );
+
+    // Update styles
     l(`row${index}`).style.pointerEvents = 'none';
     l(`row${index}`).style.opacity = '0.4';
     l(`productLock${index}`).innerHTML = 'Unlock';
     l(`productLock${index}`).style.pointerEvents = 'auto';
   } else {
-    Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames =
-      Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames.filter(
-        (value) => value !== index.toString(),
-      );
+    // Remove from storing array
+    if (
+      Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames.includes(
+        index.toString(),
+      )
+    ) {
+      Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames =
+        Game.mods.cookieMonsterFramework.saveData.cookieMonsterMod.lockedMinigames.filter(
+          (value) => value !== index.toString(),
+        );
+    }
+
+    // Update styles
     l(`productLock${index}`).innerHTML = 'Lock';
     l(`row${index}`).style.pointerEvents = 'auto';
     l(`row${index}`).style.opacity = '1';