diff --git a/src/main/kotlin/ru/dbotthepony/mc/otm/container/UpgradeContainer.kt b/src/main/kotlin/ru/dbotthepony/mc/otm/container/UpgradeContainer.kt
index 0c0070806..78464971c 100644
--- a/src/main/kotlin/ru/dbotthepony/mc/otm/container/UpgradeContainer.kt
+++ b/src/main/kotlin/ru/dbotthepony/mc/otm/container/UpgradeContainer.kt
@@ -1,5 +1,6 @@
 package ru.dbotthepony.mc.otm.container
 
+import net.minecraft.world.item.ItemStack
 import ru.dbotthepony.mc.otm.capability.IMatteryUpgrade
 import ru.dbotthepony.mc.otm.capability.MatteryCapability
 import ru.dbotthepony.mc.otm.capability.UpgradeType
@@ -15,8 +16,13 @@ class UpgradeContainer(
 	slotCount: Int,
 	val allowedUpgrades: Set<UpgradeType> = UpgradeType.ALL,
 	val shouldLockUpgradeSlots: BooleanSupplier = BooleanSupplier { false },
-	listener: Runnable = Runnable {}
-) : MatteryContainer(listener, slotCount), IMatteryUpgrade {
+	private val listener: Runnable = Runnable {}
+) : EnhancedContainer(slotCount), IMatteryUpgrade {
+	override fun notifySlotChanged(slot: Int, old: ItemStack) {
+		super.notifySlotChanged(slot, old)
+		listener.run()
+	}
+
 	override val upgradeTypes: Set<UpgradeType>
 		get() = setOf()