diff --git a/src/main/java/ru/bitheaven/createairfabric/mixin/EntityOxygenSystemMixin.java b/src/main/java/ru/bitheaven/createairfabric/mixin/EntityOxygenSystemMixin.java
index c89197c..11943bd 100644
--- a/src/main/java/ru/bitheaven/createairfabric/mixin/EntityOxygenSystemMixin.java
+++ b/src/main/java/ru/bitheaven/createairfabric/mixin/EntityOxygenSystemMixin.java
@@ -19,9 +19,10 @@ public abstract class EntityOxygenSystemMixin {
     @Redirect(at = @At(value = "INVOKE", target = "Learth/terrarium/adastra/common/items/armor/SpaceSuitItem;hasOxygen(Lnet/minecraft/world/entity/Entity;)Z"),
             method = "entityTick(Lnet/minecraft/server/level/ServerLevel;Lnet/minecraft/world/entity/LivingEntity;)V")
     private boolean redirectOxygenTick(Entity entity) {
+        if (!(entity instanceof LivingEntity livingEntity)) return SpaceSuitItem.hasOxygen(entity);
         boolean createOxygen = true;
-        createOxygen &= !DivingHelmetItem.getWornItem(entity).isEmpty();
-        createOxygen &= !BacktankUtil.getAllWithAir((LivingEntity) entity).isEmpty();
-        return SpaceSuitItem.hasOxygen(entity) || createOxygen;
+        createOxygen &= !DivingHelmetItem.getWornItem(livingEntity).isEmpty();
+        createOxygen &= !BacktankUtil.getAllWithAir(livingEntity).isEmpty();
+        return SpaceSuitItem.hasOxygen(livingEntity) || createOxygen;
     }
 }