diff --git a/kubejs/startup_scripts/CustomAdditions.js b/kubejs/startup_scripts/CustomAdditions.js
index 515e6808b..d3bd46747 100644
--- a/kubejs/startup_scripts/CustomAdditions.js
+++ b/kubejs/startup_scripts/CustomAdditions.js
@@ -61,8 +61,8 @@ global.iceAndFirePearls = (/** @type {$ServerLevel_}} */ level, /** @type {$Play
     let mainItemStack = player.getItemInHand(hand)
     if (level.clientSide) return true
     // currentStopwatch = $Stopwatch.createStarted()
-    let $DragonType = Java.loadClass("com.iafenvoy.iceandfire.data.DragonType")
-    let $EntityDragonBase = Java.loadClass("com.iafenvoy.iceandfire.entity.EntityDragonBase")
+    let $DragonType = Java.loadClass("com.iafenvoy.iceandfire.registry.IafDragonTypes")
+    let $EntityDragonBase = Java.loadClass("com.iafenvoy.iceandfire.entity.DragonBaseEntity")
     let $AABB = Java.loadClass("net.minecraft.world.phys.AABB")
     let $Mth = Java.loadClass("net.minecraft.util.Mth")
     let $EyeOfEnder = Java.loadClass("net.minecraft.world.entity.projectile.EyeOfEnder")