diff --git a/src/data/kotlin/ru/dbotthepony/mc/otm/datagen/loot/LootModifiersData.kt b/src/data/kotlin/ru/dbotthepony/mc/otm/datagen/loot/LootModifiersData.kt
index 45a9c5d40..59aba233f 100644
--- a/src/data/kotlin/ru/dbotthepony/mc/otm/datagen/loot/LootModifiersData.kt
+++ b/src/data/kotlin/ru/dbotthepony/mc/otm/datagen/loot/LootModifiersData.kt
@@ -145,4 +145,14 @@ fun addLootModifiers(it: LootModifiers) {
 			ItemStack(MItems.EXOSUIT_PROBE)
 		)
 	))
+
+	it.add("wither_exosuit_upgrades", LootTableSeparatedAppender(
+		arrayOf(LootTableIdCondition(EntityType.WITHER.defaultLootTable)),
+		ItemStack(MItems.ExosuitUpgrades.INVENTORY_UPGRADE_WITHER, 1) to 1.0,
+	))
+
+	it.add("ender_dragon_exosuit_upgrades", LootTableSeparatedAppender(
+		arrayOf(LootTableIdCondition(EntityType.ENDER_DRAGON.defaultLootTable)),
+		ItemStack(MItems.ExosuitUpgrades.INVENTORY_UPGRADE_ENDER_DRAGON, 1) to 1.0,
+	))
 }