diff --git a/src/main/java/dev/micle/loginprotection/network/client/InputPacket.java b/src/main/java/dev/micle/loginprotection/network/client/InputPacket.java
index be48853..f98241c 100755
--- a/src/main/java/dev/micle/loginprotection/network/client/InputPacket.java
+++ b/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());
             }
         });