1 files changed, 5 insertions, 6 deletions
diff --git a/src/main/java/me/shedaniel/mixins/MixinDoneLoading.java b/src/main/java/me/shedaniel/mixins/MixinDoneLoading.java
index df79245db..c3b4ff186 100755
--- a/src/main/java/me/shedaniel/mixins/MixinDoneLoading.java
+++ b/src/main/java/me/shedaniel/mixins/MixinDoneLoading.java
@@ -1,8 +1,8 @@
 package me.shedaniel.mixins;
 
+import me.shedaniel.Core;
 import me.shedaniel.listenerdefinitions.DoneLoading;
-import net.minecraft.init.Bootstrap;
-import org.dimdev.riftloader.RiftLoader;
+import net.minecraft.Bootstrap;
 import org.spongepowered.asm.mixin.Mixin;
 import org.spongepowered.asm.mixin.injection.At;
 import org.spongepowered.asm.mixin.injection.Inject;
@@ -13,10 +13,9 @@ import org.spongepowered.asm.mixin.injection.callback.CallbackInfo;
  */
 @Mixin(Bootstrap.class)
 public class MixinDoneLoading {
-    @Inject(method = "register", at = @At("RETURN"))
+    @Inject(method = "initialize", at = @At("RETURN"))
     private static void onBootstrapRegister(CallbackInfo ci) {
-        for(DoneLoading listener : RiftLoader.instance.getListeners(DoneLoading.class)) {
-            listener.onDoneLoading();
-        }
+        System.out.println("Done Loading");
+        Core.getListeners(DoneLoading.class).forEach(DoneLoading::onDoneLoading);
     }
 }