diff --git a/src/main/java/ru/bitheaven/createairfabric/CreateAirFabric.java b/src/main/java/ru/bitheaven/createairfabric/CreateAirFabric.java
index 5427e3a..073a0fd 100644
--- a/src/main/java/ru/bitheaven/createairfabric/CreateAirFabric.java
+++ b/src/main/java/ru/bitheaven/createairfabric/CreateAirFabric.java
@@ -5,7 +5,7 @@ import earth.terrarium.adastra.api.systems.OxygenApi;
 import fuzs.thinair.helper.AirQualityHelperImpl;
 import net.fabricmc.api.ModInitializer;
 import net.fabricmc.loader.api.FabricLoader;
-import net.minecraft.world.entity.Entity;
+import net.minecraft.world.entity.LivingEntity;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 
@@ -19,7 +19,7 @@ public class CreateAirFabric implements ModInitializer {
         LOGGER.info("Create addon mod [{}] is loading alongside Create [{}]!", NAME, Create.VERSION);
     }
 
-    public static boolean isOxygen(Entity entity) {
+    public static boolean isOxygen(LivingEntity entity) {
         boolean oxygen = true;
         if (FabricLoader.getInstance().isModLoaded("thinair")) {
             oxygen &= new AirQualityHelperImpl()