Fixed a bug with cancelAirIncrement

common/src/main/java/draylar/identity/mixin/LivingEntityMixin.java

@@ -61,25 +61,6 @@ private void onDeath(DamageSource source, CallbackInfo ci) {
         }
     }
 
-    @Redirect(
-            method = "baseTick",
-            at = @At(value = "INVOKE", target = "Lnet/minecraft/entity/LivingEntity;setAir(I)V", ordinal = 2)
-    )
-    private void cancelAirIncrement(LivingEntity livingEntity, int air) {
-        // Aquatic creatures should not regenerate breath on land
-        if ((Object) this instanceof PlayerEntity player) {
-            LivingEntity identity = PlayerIdentity.getIdentity(player);
-
-            if (identity != null) {
-                if (Identity.isAquatic(identity)) {
-                    return;
-                }
-            }
-        }
-
-        this.setAir(this.getNextAirOnLand(this.getAir()));
-    }
-
     @Redirect(
             method = "travel",
             at = @At(value = "INVOKE", target = "Lnet/minecraft/entity/LivingEntity;hasStatusEffect(Lnet/minecraft/entity/effect/StatusEffect;)Z", ordinal = 0)

fabric/src/main/java/draylar/identity/fabric/mixin/LivingEntityMixin.java

@@ -0,0 +1,40 @@
+package draylar.identity.mixin;
+
+import draylar.identity.Identity;
+import draylar.identity.api.PlayerIdentity;
+import net.minecraft.entity.*;
+import net.minecraft.world.World;
+import org.spongepowered.asm.mixin.Mixin;
+import org.spongepowered.asm.mixin.Shadow;
+import org.spongepowered.asm.mixin.injection.At;
+import org.spongepowered.asm.mixin.injection.Redirect;
+
+@Mixin(LivingEntity.class)
+public abstract class LivingEntityMixin extends Entity {
+
+    @Shadow
+    protected abstract int getNextAirOnLand(int air);
+
+    protected LivingEntityMixin(EntityType<?> type, World world) {
+        super(type, world);
+    }
+
+    @Redirect(
+            method = "baseTick",
+            at = @At(value = "INVOKE", target = "Lnet/minecraft/entity/LivingEntity;setAir(I)V", ordinal = 2)
+    )
+    private void cancelAirIncrement(LivingEntity livingEntity, int air) {
+        // Aquatic creatures should not regenerate breath on land
+        if ((Object) this instanceof PlayerEntity player) {
+            LivingEntity identity = PlayerIdentity.getIdentity(player);
+
+            if (identity != null) {
+                if (Identity.isAquatic(identity)) {
+                    return;
+                }
+            }
+        }
+
+        this.setAir(this.getNextAirOnLand(this.getAir()));
+    }
+}

fabric/src/main/resources/identity-fabric.mixins.json

@@ -4,6 +4,7 @@
   "package": "draylar.identity.fabric.mixin",
   "compatibilityLevel": "JAVA_16",
   "mixins": [
+    "LivingEntityMixin",
     "WitherEntityMixin"
   ],
   "client": [

forge/src/main/java/draylar/identity/forge/WalkersForgeEventHandler.java

@@ -0,0 +1,27 @@
+package tocraft.walkers.forge;
+
+import net.minecraft.entity.*;
+import net.minecraft.entity.player.PlayerEntity;
+import net.minecraftforge.event.entity.living.LivingBreatheEvent;
+import net.minecraftforge.eventbus.api.SubscribeEvent;
+import net.minecraftforge.fml.common.Mod;
+import draylar.identity.Identity;
+import draylar.identity.api.PlayerIdentity;
+
+@Mod.EventBusSubscriber(modid = "identity")
+public class WalkersForgeEventHandler {
+
+	@SubscribeEvent
+	public static void livingBreath(LivingBreatheEvent event) {
+		if (event.getEntity() instanceof PlayerEntity) {
+			LivingEntity identity = PlayerIdentity.getIdentity((PlayerEntity) event.getEntity());
+
+			if (identity != null) {
+				if (Identity.isAquatic(identity)) {
+					event.setCanBreathe(false);
+
+				}
+			}
+		}
+	}
+}

gradle.properties

@@ -10,7 +10,7 @@ maven_group=dev.draylar
 fabric_loader_version=0.14.21
 fabric_api_version=0.84.0+1.20.1
 yarn_mappings=1.20.1+build.8
-forge_version=47.0.43
+forge_version=47.1.43
 
 # Dependency Versions
 architectury_version=9.0.8