More events

1 files changed, 3 insertions, 4 deletions

diff --git a/src/main/java/net/elytrium/limboauth/handler/AuthSessionHandler.java b/src/main/java/net/elytrium/limboauth/handler/AuthSessionHandler.java
index 9a6ca8b..43895f2 100644
--- a/src/main/java/net/elytrium/limboauth/handler/AuthSessionHandler.java
+++ b/src/main/java/net/elytrium/limboauth/handler/AuthSessionHandler.java
@@ -39,7 +39,6 @@ import net.elytrium.limboapi.api.player.LimboPlayer;
 import net.elytrium.limboauth.LimboAuth;
 import net.elytrium.limboauth.Settings;
 import net.elytrium.limboauth.event.PostAuthorizationEvent;
-import net.elytrium.limboauth.event.PostEvent;
 import net.elytrium.limboauth.event.PostRegisterEvent;
 import net.elytrium.limboauth.event.TaskEvent;
 import net.elytrium.limboauth.migration.MigrationHash;
@@ -316,17 +315,17 @@ public class AuthSessionHandler implements LimboSessionHandler {
     }
 
     this.plugin.getServer().getEventManager()
-        .fire(new PostAuthorizationEvent(this.player, this.playerInfo))
+        .fire(new PostAuthorizationEvent(this.player, this.playerInfo, this::finishAuth))
         .thenAcceptAsync(this::finishAuth);
   }
 
   private void finishRegister() {
     this.plugin.getServer().getEventManager()
-        .fire(new PostRegisterEvent(this.player, this.playerInfo))
+        .fire(new PostRegisterEvent(this.player, this.playerInfo, this::finishAuth))
         .thenAcceptAsync(this::finishAuth);
   }
 
-  private void finishAuth(PostEvent event) {
+  private void finishAuth(TaskEvent event) {
     if (Settings.IMP.MAIN.CRACKED_TITLE_SETTINGS.CLEAR_AFTER_LOGIN) {
       this.proxyPlayer.clearTitle();
     }