diff --git a/project/SPT.SinglePlayer/Patches/MainMenu/RemoveStashUpgradeLabelPatch.cs b/project/SPT.SinglePlayer/Patches/MainMenu/RemoveStashUpgradeLabelPatch.cs index bfebaa3..c1250d5 100644 --- a/project/SPT.SinglePlayer/Patches/MainMenu/RemoveStashUpgradeLabelPatch.cs +++ b/project/SPT.SinglePlayer/Patches/MainMenu/RemoveStashUpgradeLabelPatch.cs @@ -13,13 +13,16 @@ public class RemoveStashUpgradeLabelPatch : ModulePatch { protected override MethodBase GetTargetMethod() { - // InventoryScreen has 2 Show methods - return AccessTools.FirstMethod(typeof(InventoryScreen), info => info.Name == "Show"); + return typeof(InventoryScreen).GetMethod(nameof(InventoryScreen.Awake)); } [PatchPostfix] public static void Postfix(SimpleStashPanel __instance) { - Object.Destroy(__instance.transform.Find("Items Panel/Stash Panel/Simple Panel/Header/ExternalObtain").gameObject); + GameObject externalObtain = __instance.transform.Find("Items Panel/Stash Panel/Simple Panel/Header/ExternalObtain").gameObject; + if (externalObtain != null) + { + Object.Destroy(externalObtain); + } } } \ No newline at end of file