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);
                 }
             }
         }