Fix fly bug on WorldChange

MAStaff-Bukkit/src/main/java/es/angelillo15/mast/bukkit/MAStaff.java

@@ -176,6 +176,7 @@ public void registerListeners() {
     registerListener(injector.getInstance(OnAttack.class));
     registerListener(injector.getInstance(OnDamage.class));
     registerListener(injector.getInstance(CommandManagerHandler.class));
+    registerListener(injector.getInstance(OnWorldChange.class));
     injector.getInstance(EventManager.class).registerListener(injector.getInstance(OnAddonDisable.class));
 
     if (Config.Freeze.enabled()) registerListener(injector.getInstance(FreezeListener.class));

MAStaff-Bukkit/src/main/java/es/angelillo15/mast/bukkit/listener/staffmode/OnWorldChange.java

@@ -0,0 +1,29 @@
+package es.angelillo15.mast.bukkit.listener.staffmode;
+
+import com.google.inject.Inject;
+import com.nookure.mast.api.event.BukkitListener;
+import es.angelillo15.mast.api.IStaffPlayer;
+import es.angelillo15.mast.api.managers.StaffManager;
+import es.angelillo15.mast.bukkit.MAStaff;
+import org.bukkit.Bukkit;
+import org.bukkit.entity.Player;
+import org.bukkit.event.EventHandler;
+import org.bukkit.event.player.PlayerChangedWorldEvent;
+
+public class OnWorldChange implements BukkitListener<PlayerChangedWorldEvent> {
+  @Inject
+  private StaffManager staffManager;
+  @Override
+  @EventHandler
+  public void handle(PlayerChangedWorldEvent event) {
+    Player player = event.getPlayer();
+    IStaffPlayer staffPlayer = staffManager.getStaffPlayer(player);
+
+    if (staffPlayer == null) return;
+    if (!staffPlayer.isStaffMode()) return;
+
+    Bukkit.getScheduler().runTaskLater(MAStaff.getPlugin(), () -> {
+      player.setAllowFlight(true);
+    }, 20L);
+  }
+}