diff --git a/neoforged/src/main/java/pl/panszelescik/colorize/neoforge/ColorizeNeoForge.java b/neoforged/src/main/java/pl/panszelescik/colorize/neoforge/ColorizeNeoForge.java
index 9b72c47..c7b884a 100644
--- a/neoforged/src/main/java/pl/panszelescik/colorize/neoforge/ColorizeNeoForge.java
+++ b/neoforged/src/main/java/pl/panszelescik/colorize/neoforge/ColorizeNeoForge.java
@@ -1,9 +1,9 @@
 package pl.panszelescik.colorize.neoforge;
 
-import net.neoforged.bus.api.IEventBus;
 import net.neoforged.fml.ModContainer;
 import net.neoforged.fml.common.Mod;
 import net.neoforged.fml.config.ModConfig;
+import net.neoforged.neoforge.common.NeoForge;
 import net.neoforged.neoforge.event.entity.player.PlayerInteractEvent;
 import org.jetbrains.annotations.NotNull;
 
@@ -13,8 +13,8 @@ public class ColorizeNeoForge {
     public static final String MODID = "colorize";
     private final ColorizeNeoForgeHandler handler = new ColorizeNeoForgeHandler(ColorizeNeoForgeConfig.CONFIG);
 
-    public ColorizeNeoForge(ModContainer modContainer, IEventBus bus) {
-        bus.addListener(this::onRightClickBlock);
+    public ColorizeNeoForge(ModContainer modContainer) {
+        NeoForge.EVENT_BUS.addListener(this::onRightClickBlock);
 
         modContainer.registerConfig(ModConfig.Type.COMMON, ColorizeNeoForgeConfig.SPEC);
     }