diff --git a/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt b/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
index bbe19aa..eca15bf 100644
--- a/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
+++ b/neoforge/src/main/kotlin/fr/username404/snowygui/forge/ForgeInit.kt
@@ -9,6 +9,7 @@ import net.neoforged.fml.ModList
 import net.neoforged.fml.common.Mod
 import net.neoforged.fml.event.lifecycle.FMLClientSetupEvent
 import net.neoforged.neoforge.client.gui.IConfigScreenFactory
+import net.neoforged.neoforge.common.NeoForge
 import net.neoforged.neoforgespi.language.ModFileScanData
 import org.objectweb.asm.Type
 
@@ -27,7 +28,7 @@ class ForgeInit(container: ModContainer): Snowy() {
         .map { Class.forName((it!!.javaClass.getDeclaredField("clazz").apply { isAccessible = true }.get(it) as Type).className).asSubclass(ButtonImpl::class.java)}
         .toSet()
     init {
-        container.eventBus!!.run {
+        NeoForge.EVENT_BUS.run {
             addListener(this@ForgeInit::initSetup)
             register(EndTickHandler)
             register(MiscHandlers)