From 31c3c4c981d0c5dd570195921ee1ae646f7db700 Mon Sep 17 00:00:00 2001
From: baz <bazzadobs@live.co.uk>
Date: Fri, 25 Jul 2025 23:43:53 +0100
Subject: [PATCH] Fix bugs

---
 Source/vampires/Widgets/StarterWeaponButtonDataObject.cpp | 7 ++-----
 1 file changed, 2 insertions(+), 5 deletions(-)

diff --git a/Source/vampires/Widgets/StarterWeaponButtonDataObject.cpp b/Source/vampires/Widgets/StarterWeaponButtonDataObject.cpp
index 067a01a..cdfe2b6 100644
--- a/Source/vampires/Widgets/StarterWeaponButtonDataObject.cpp
+++ b/Source/vampires/Widgets/StarterWeaponButtonDataObject.cpp
@@ -8,14 +8,10 @@
 void UStarterWeaponButtonDataObject::SetData(AWeapon* Weapon, UUserWidget* parent)
 {
 	WeaponName = Weapon->GetWeaponName();
+	Description = Weapon->GetDescription();
 	WeaponIcon = Weapon->GetIcon();
 	WeaponInstance = Weapon;
 	Parent = parent;
-	
-	if (Weapon->GetUpgradeDescriptions().Num() > Weapon->GetWeaponLevel())
-	{
-		Description = Weapon->GetUpgradeDescriptions()[Weapon->GetWeaponLevel()];
-	}
 }
 
 void UStarterWeaponButtonDataObject::SetData(TSubclassOf<AWeapon> Weapon, UUserWidget* parent)
@@ -23,6 +19,7 @@ void UStarterWeaponButtonDataObject::SetData(TSubclassOf<AWeapon> Weapon, UUserW
 	if (AWeapon* tempWeapon = NewObject<AWeapon>(this, Weapon))
 	{
 		SetData(tempWeapon, parent);
+		WeaponTemplate = Weapon;
 	}
 }