Note: Im not him, + helper

Author: Tofaa2

api/src/main/java/me/tofaa/entitylib/meta/EntityMeta.java

@@ -25,6 +25,7 @@ public class EntityMeta implements EntityMetadataProvider {
 
     static final MetaConverterRegistry META_CONVERTER_REGISTRY = new MetaConverterRegistry();
 
+
     public static @NotNull BiFunction<Integer, Metadata, EntityMeta> getConverter(EntityType entityType) {
         return META_CONVERTER_REGISTRY.get(entityType);
     }
@@ -144,16 +145,16 @@ public void setFlyingWithElytra(boolean value) {
     }
 
     public short getAirTicks() {
-        // Food for thought later on 
-        // if (isVersionNewer(ServerVersion.V_26_1)) {
-        //     return this.metadata.getIndex((byte)1, (short)300);
-        // }
+        // Food for thought later on
+        if (isVersion(ServerVersion.V_1_21_11, VersionComparison.NEWER_THAN)) {
+            return (short)this.metadata.getIndex((byte)1, (int)300).shortValue();
+        }
         return this.metadata.getIndex((byte)1, (short) 300);
     }
 
     public void setAirTicks(short value) {
-        if (isVersionNewer(ServerVersion.V_26_1)) {
-            this.metadata.setIndex((byte)1, EntityDataTypes.INT, value);
+        if (isVersion(ServerVersion.V_1_21_11, VersionComparison.NEWER_THAN)) {
+            this.metadata.setIndex((byte)1, EntityDataTypes.INT, (int)value);
         }
         else {
             this.metadata.setIndex((byte)1, EntityDataTypes.SHORT, value);

api/src/main/java/me/tofaa/entitylib/wrapper/WrapperEntity.java

@@ -1,6 +1,7 @@
 package me.tofaa.entitylib.wrapper;
 
 import com.github.retrooper.packetevents.protocol.entity.type.EntityType;
+import com.github.retrooper.packetevents.protocol.entity.type.EntityTypes;
 import com.github.retrooper.packetevents.protocol.player.User;
 import com.github.retrooper.packetevents.protocol.world.Location;
 import com.github.retrooper.packetevents.util.Vector3d;
@@ -39,6 +40,21 @@
 import org.jetbrains.annotations.UnmodifiableView;
 
 public class WrapperEntity implements Tickable {
+    
+    public static Class<? extends WrapperEntity> getWrapperClassFromType(EntityType e) {
+        if (e == EntityTypes.PLAYER) {
+            return WrapperPlayer.class;
+        }
+        if (EntityMeta.isLivingEntity(e)) {
+            return WrapperLivingEntity.class;
+        }
+        if (e == EntityTypes.EXPERIENCE_ORB) {
+            return WrapperExperienceOrbEntity.class;
+        }
+        return WrapperEntity.class;
+    }
+
+
     private final UUID uuid;
     private final int entityId;
     private final EntityType entityType;