diff --git a/Directory.Build.props b/Directory.Build.props
index a0afa3ac..d0100ed4 100644
--- a/Directory.Build.props
+++ b/Directory.Build.props
@@ -1,6 +1,6 @@
 <Project>
   <PropertyGroup>
-    <Version>1.7.1.0</Version>
+    <Version>1.7.2.0</Version>
     <GameVersion>1.5.5</GameVersion>
     <GameBranch>Beta</GameBranch>
     <AdditionalInfo></AdditionalInfo>
diff --git a/Patches/Party/FreeTroopUpgradesPatch.cs b/Patches/Party/FreeTroopUpgradesPatch.cs
index fbe55401..54a75132 100644
--- a/Patches/Party/FreeTroopUpgradesPatch.cs
+++ b/Patches/Party/FreeTroopUpgradesPatch.cs
@@ -1,22 +1,17 @@
 using BannerlordCheats.Settings;
 using HarmonyLib;
-using System.Linq;
 using TaleWorlds.CampaignSystem;
-using TaleWorlds.CampaignSystem.SandBox.GameComponents.Party;
+using TaleWorlds.CampaignSystem.SandBox.GameComponents;
 
 namespace BannerlordCheats.Patches
 {
-    [HarmonyPatch(typeof(DefaultPartyWageModel), nameof(DefaultPartyWageModel.GetGoldCostForUpgrade))]
+    [HarmonyPatch(typeof(DefaultPartyTroopUpgradeModel), nameof(DefaultPartyTroopUpgradeModel.GetGoldCostForUpgrade))]
     public static class FreeTroopUpgradesPatch
     {
         [HarmonyPostfix]
-        public static void GetGoldCostForUpgrade(CharacterObject characterObject, CharacterObject upgradeTarget, ref int __result)
+        public static void GetGoldCostForUpgrade(ref PartyBase party, ref CharacterObject characterObject, ref CharacterObject upgradeTarget, ref int __result)
         {
-            var playerTeam = PartyBase.MainParty?.MemberRoster?.Select(x => x.Character?.Id).ToArray();
-
-            if (playerTeam != null
-                && characterObject?.Id != null
-                && (playerTeam.Contains(characterObject?.Id))
+            if ((party?.Leader?.IsPlayerCharacter ?? false)
                 && BannerlordCheatsSettings.Instance.FreeTroopUpgrades)
             {
                 __result = 0;