diff --git a/src/client/java/tobinio/bettersnowcoverage/BetterSnowChecker.java b/src/client/java/tobinio/bettersnowcoverage/BetterSnowChecker.java
index 9ce82d6..3b4c533 100644
--- a/src/client/java/tobinio/bettersnowcoverage/BetterSnowChecker.java
+++ b/src/client/java/tobinio/bettersnowcoverage/BetterSnowChecker.java
@@ -2,13 +2,13 @@ package tobinio.bettersnowcoverage;
 
 import net.minecraft.block.*;
 import net.minecraft.client.MinecraftClient;
-import net.minecraft.client.network.ClientPlayerEntity;
+import net.minecraft.client.world.ClientWorld;
 import net.minecraft.registry.tag.BlockTags;
 import net.minecraft.registry.tag.TagKey;
 import net.minecraft.state.property.Properties;
 import net.minecraft.util.math.BlockPos;
 import net.minecraft.util.math.Direction;
-import net.minecraft.world.BlockRenderView;
+import net.minecraft.world.Heightmap;
 import tobinio.bettersnowcoverage.config.Config;
 
 /**
@@ -29,17 +29,17 @@ public class BetterSnowChecker {
         NONE, WITH_LAYER, WITHOUT_LAYER,
     }
 
-    public static SnowState shouldHaveSnow(BlockPos pos, BlockRenderView world) {
-        ClientPlayerEntity player = MinecraftClient.getInstance().player;
+    public static SnowState shouldHaveSnow(BlockPos pos) {
+        ClientWorld world = MinecraftClient.getInstance().world;
 
-        if (player == null) {
+        if (world == null) {
             BetterSnowCoverage.LOGGER.warn("no player found");
             return SnowState.NONE;
         }
 
         var state = world.getBlockState(pos);
 
-        if (state.isSideSolidFullSquare(world, pos, Direction.DOWN) || state.getBlock() == Blocks.AIR) {
+        if (state.isSideSolidFullSquare(world, pos, Direction.DOWN) || state.isIn(BlockTags.AIR)) {
             return SnowState.NONE;
         }
 
@@ -74,15 +74,38 @@ public class BetterSnowChecker {
         return false;
     }
 
-    private static boolean hasSnowNeighbor(BlockPos pos, BlockRenderView world) {
+    private static boolean hasSnowNeighbor(BlockPos pos, ClientWorld world) {
         var directions = new Direction[]{Direction.NORTH, Direction.EAST, Direction.SOUTH, Direction.WEST};
 
         for (Direction direction : directions) {
-            BlockState neighbor = world.getBlockState(pos.add(direction.getOffsetX(), 0, direction.getOffsetZ()));
+            if (!hasSnowInDirection(pos, world, direction)) {
+                return false;
+            }
+        }
+
+        return true;
+    }
+
+    private static boolean hasSnowInDirection(BlockPos pos, ClientWorld world, Direction direction) {
+        var maxDepth = 8;
+
+        while (maxDepth-- > 0) {
+            pos = pos.add(direction.getOffsetX(), 0, direction.getOffsetZ());
+            pos = new BlockPos(pos.getX(), world.getTopY(Heightmap.Type.MOTION_BLOCKING, pos.getX(), pos.getZ()) - 1, pos.getZ());
 
-            if (neighbor.isIn(BlockTags.SNOW)) {
+            var maxDepth2 = 8;
+
+            while (!world.getBlockState(pos).isFullCube(world, pos) && maxDepth2-- > 0) {
+                pos = pos.down();
+            }
+
+            if (world.getBlockState(pos.up()).isIn(BlockTags.SNOW)) {
                 return true;
             }
+
+            if (world.getBlockState(pos.up()).isIn(BlockTags.AIR)) {
+                return false;
+            }
         }
 
         return false;
diff --git a/src/client/java/tobinio/bettersnowcoverage/mixin/client/SectionBuilderMixin.java b/src/client/java/tobinio/bettersnowcoverage/mixin/client/SectionBuilderMixin.java
index 1f03bdf..348ebcf 100644
--- a/src/client/java/tobinio/bettersnowcoverage/mixin/client/SectionBuilderMixin.java
+++ b/src/client/java/tobinio/bettersnowcoverage/mixin/client/SectionBuilderMixin.java
@@ -40,7 +40,7 @@ public class SectionBuilderMixin {
             @Local BlockState blockState, @Local (ordinal = 2) BlockPos blockPos, @Local MatrixStack matrixStack,
             @Local BufferBuilder bufferBuilder, @Local Random random) {
 
-        var snowState = BetterSnowChecker.shouldHaveSnow(blockPos.up(), renderRegion);
+        var snowState = BetterSnowChecker.shouldHaveSnow(blockPos.up());
 
         if (snowState == BetterSnowChecker.SnowState.WITH_LAYER) {
             matrixStack.push();
@@ -51,9 +51,8 @@ public class SectionBuilderMixin {
     }
 
     @ModifyVariable (method = "build", at = @At ("STORE"), ordinal = 0)
-    private BlockState setGrassState(BlockState state, @Local (ordinal = 2) BlockPos blockPos,
-            @Local (argsOnly = true) ChunkRendererRegion renderRegion) {
-        var snowState = BetterSnowChecker.shouldHaveSnow(blockPos.up(), renderRegion);
+    private BlockState setGrassState(BlockState state, @Local (ordinal = 2) BlockPos blockPos) {
+        var snowState = BetterSnowChecker.shouldHaveSnow(blockPos.up());
 
         if (snowState != BetterSnowChecker.SnowState.NONE) {
             return BetterSnowChecker.getSnowState(state);
diff --git a/src/client/java/tobinio/bettersnowcoverage/mixin/client/sodium/ChunkBuilderMeshingTaskMixin.java b/src/client/java/tobinio/bettersnowcoverage/mixin/client/sodium/ChunkBuilderMeshingTaskMixin.java
index b7b3962..8007c53 100644
--- a/src/client/java/tobinio/bettersnowcoverage/mixin/client/sodium/ChunkBuilderMeshingTaskMixin.java
+++ b/src/client/java/tobinio/bettersnowcoverage/mixin/client/sodium/ChunkBuilderMeshingTaskMixin.java
@@ -31,7 +31,7 @@ public class ChunkBuilderMeshingTaskMixin {
             CallbackInfoReturnable<ChunkBuildOutput> cir, @Local BlockRenderContext ctx,
             @Local ChunkBuildBuffers buffers, @Local BlockRenderCache cache, @Local WorldSlice slice,
             @Local (ordinal = 1) BlockPos.Mutable modelOffset) {
-        BetterSnowChecker.SnowState snowState = BetterSnowChecker.shouldHaveSnow(ctx.pos().up(), ctx.world());
+        BetterSnowChecker.SnowState snowState = BetterSnowChecker.shouldHaveSnow(ctx.pos().up());
 
         if (snowState != BetterSnowChecker.SnowState.NONE) {
             var newState = BetterSnowChecker.getSnowState(ctx.state());