InputPacket: Instead of checking for any state other than ACTIVE, it will check for the JOINING or AFK state before updating the state.

src/main/java/dev/micle/loginprotection/network/client/InputPacket.java

@@ -34,8 +34,9 @@ public class InputPacket {
                 return;
             }
 
-            // Update player state if they are not active
-            if (!protectedPlayer.getState().equals(ProtectedPlayer.State.ACTIVE)) {
+            // Update player state if they are joining or afk
+            if (protectedPlayer.getState().equals(ProtectedPlayer.State.JOINING) ||
+                    protectedPlayer.getState().equals(ProtectedPlayer.State.AFK)) {
                 ProtectedPlayerManager.updateState(protectedPlayer.getPlayerUUID());
             }
         });