diff --git a/src/controllers/api/upgradesController.ts b/src/controllers/api/upgradesController.ts
index 568a7cf4..5947edb1 100644
--- a/src/controllers/api/upgradesController.ts
+++ b/src/controllers/api/upgradesController.ts
@@ -108,7 +108,7 @@ export const upgradesController: RequestHandler = async (req, res) => {
                             arr[arr.indexOf(upgrade.ItemId.$id)] = "";
                         }
                         // We need to convert RawUpgrade into Upgrade once it's attached
-                        const rawUpgrade = inventory.RawUpgrades.find(x => x.ItemType == upgrade.ItemType);
+                        const rawUpgrade = inventory.RawUpgrades.id(upgrade.ItemId.$id);
                         if (rawUpgrade) {
                             const newId = new Types.ObjectId().toString();
                             arr[upgrade.Slot - 1] = newId;