From da5add33e8d5e7d11882bbabbd5cfad2547f610f Mon Sep 17 00:00:00 2001
From: garutako <adam-hamid.salah-salah.1@ens.etsmtl.ca>
Date: Mon, 13 Oct 2025 00:32:40 -0400
Subject: [PATCH] fixed house upgrade bug

---
 Assets/Scripts/Ally/AllyUpgrade.cs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Assets/Scripts/Ally/AllyUpgrade.cs b/Assets/Scripts/Ally/AllyUpgrade.cs
index e618912..1de70ec 100644
--- a/Assets/Scripts/Ally/AllyUpgrade.cs
+++ b/Assets/Scripts/Ally/AllyUpgrade.cs
@@ -48,10 +48,10 @@ public class AllyUpgrade : MonoBehaviour, IPointerClickHandler
         {
             if (_upgradeList.Count > i && _upgradeList[i] != null)
             {
-                GameObject upgradeEnfant = _instance.transform.Find("Upgrade" + (i + 1)).gameObject;
-                if (upgradeEnfant.TryGetComponent<UpgradePlacementButton>(out UpgradePlacementButton button))
+                Transform upgradeEnfant = _instance.transform.Find("Upgrade" + (i + 1));
+                if (upgradeEnfant && upgradeEnfant.TryGetComponent<UpgradePlacementButton>(out UpgradePlacementButton button))
                 {
-                    button.Initialize(_upgradeList[i], gameObject, canvas);
+                    button.Initialize(_upgradeList[i], gameObject.gameObject, canvas);
                 }
             }
         }