diff --git a/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs b/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
index 4af1dbe..47b5409 100644
--- a/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
+++ b/PuzzleGameProject/Assets/Scripts/UI/ChestPopUpController.cs
@@ -11,8 +11,12 @@ namespace UI
         public event Action ArmorAndDiamondSelected;
 
         private Button _blackDie;
+        private bool _blackDieAvailable = true;
         private Button _key;
+        private bool _keyAvailable = true;
         private Button _armorAndDiamond;
+        private bool _armorAndDiamondAvailable = true;
+
         
         private void OnEnable()
         {
@@ -20,27 +24,33 @@ namespace UI
             
             _blackDie = root.Q<Button>("BlackDie");
             _blackDie.clicked += () => BlackDieSelected?.Invoke();
+            _blackDie.SetEnabled(_blackDieAvailable);
             
             _key = root.Q<Button>("Key");
             _key.clicked += () => KeySelected?.Invoke();
+            _key.SetEnabled(_keyAvailable);
             
             _armorAndDiamond = root.Q<Button>("ArmorAndDiamond");
             _armorAndDiamond.clicked += () => ArmorAndDiamondSelected?.Invoke();
+            _armorAndDiamond.SetEnabled(_armorAndDiamondAvailable);
         }
 
         public void DisableBlackDie()
         {
             _blackDie.SetEnabled(false);
+            _blackDieAvailable = false;
         }
 
         public void DisableKey()
         {
             _key.SetEnabled(false);
+            _keyAvailable = false;
         }
 
         public void DisableArmorAndDiamond()
         {
             _armorAndDiamond.SetEnabled(false);
+            _armorAndDiamondAvailable = false;
         }
     }
 }