1 files changed, 6 insertions, 3 deletions
diff --git a/src/main/java/cc/polyfrost/oneconfig/utils/TickDelay.java b/src/main/java/cc/polyfrost/oneconfig/utils/TickDelay.java
index 2fa2017..35b7b8b 100644
--- a/src/main/java/cc/polyfrost/oneconfig/utils/TickDelay.java
+++ b/src/main/java/cc/polyfrost/oneconfig/utils/TickDelay.java
@@ -5,23 +5,26 @@ import cc.polyfrost.oneconfig.events.event.Stage;
 import cc.polyfrost.oneconfig.events.event.TickEvent;
 import cc.polyfrost.oneconfig.libs.eventbus.Subscribe;
 
+/**
+ * Schedules a Runnable to be called after a certain amount of ticks.
+ */
 public class TickDelay {
     private int delay;
     private final Runnable function;
 
     public TickDelay(Runnable functionName, int ticks) {
-        EventManager.INSTANCE.getEventBus().register(this);
+        EventManager.INSTANCE.register(this);
         delay = ticks;
         function = functionName;
     }
 
     @Subscribe
-    public void onTick(TickEvent event) {
+    protected void onTick(TickEvent event) {
         if (event.stage == Stage.START) {
             // Delay expired
             if (delay < 1) {
                 function.run();
-                EventManager.INSTANCE.getEventBus().unregister(this);
+                EventManager.INSTANCE.unregister(this);
             }
             delay--;
         }