diff --git a/src/client/java/com/github/startsmercury/simply/no/shading/mixin/shading/block/sodium/FluidRendererMixin.java b/src/client/java/com/github/startsmercury/simply/no/shading/mixin/shading/block/sodium/FluidRendererMixin.java
index 78af0745..33a18fe2 100644
--- a/src/client/java/com/github/startsmercury/simply/no/shading/mixin/shading/block/sodium/FluidRendererMixin.java
+++ b/src/client/java/com/github/startsmercury/simply/no/shading/mixin/shading/block/sodium/FluidRendererMixin.java
@@ -7,7 +7,11 @@
 import com.github.startsmercury.simply.no.shading.client.Config;
 import com.github.startsmercury.simply.no.shading.client.SimplyNoShading;
 
+import me.jellysquid.mods.sodium.client.model.color.ColorProvider;
+import me.jellysquid.mods.sodium.client.model.light.LightPipeline;
+import me.jellysquid.mods.sodium.client.model.quad.ModelQuadView;
 import me.jellysquid.mods.sodium.client.render.chunk.compile.pipeline.FluidRenderer;
+import me.jellysquid.mods.sodium.client.world.WorldSlice;
 
 /**
  * The {@code FluidRendererMixin} is a {@linkplain Mixin mixin} class for the
@@ -27,7 +31,7 @@ private FluidRendererMixin() {
 	/**
 	 * This is a {@linkplain ModifyVariable variable modifier} that modifies the
 	 * parameter {@code brightness} in
-	 * {@code FluidRenderer.updateQuad(ModelQuadView, BlockAndTintGetter, BlockPos, LightPipeline, Direction, float, ColorSampler, FluidState)}.
+	 * {@code FluidRenderer.updateQuad(ModelQuadView, WorldSlice, BlockPos, LightPipeline, Direction, float, ColorProvider<FluidState>, FluidState)}.
 	 * <p>
 	 * Returns the {@code brightness} when {@linkplain Config#blockShadingEnabled
 	 * block shading is enabled}; {@code 1.0f} otherwise.
@@ -37,7 +41,7 @@ private FluidRendererMixin() {
 	 *         {@linkplain Config#blockShadingEnabled block shading is enabled};
 	 *         {@code 1.0f} otherwise
 	 */
-	@ModifyVariable(method = "updateQuad(Lme/jellysquid/mods/sodium/client/model/quad/ModelQuadView;Lnet/minecraft/world/level/BlockAndTintGetter;Lnet/minecraft/core/BlockPos;Lme/jellysquid/mods/sodium/client/model/light/LightPipeline;Lnet/minecraft/core/Direction;FLme/jellysquid/mods/sodium/client/model/quad/blender/ColorSampler;Lnet/minecraft/world/level/material/FluidState;)V",
+	@ModifyVariable(method = "updateQuad(Lme/jellysquid/mods/sodium/client/model/quad/ModelQuadView;Lme/jellysquid/mods/sodium/client/world/WorldSlice;Lnet/minecraft/core/BlockPos;Lme/jellysquid/mods/sodium/client/model/light/LightPipeline;Lnet/minecraft/core/Direction;FLme/jellysquid/mods/sodium/client/model/color/ColorProvider;Lnet/minecraft/world/level/material/FluidState;)V",
 	                at = @At("HEAD"),
 	                argsOnly = true)
 	private final float changeShade(final float brightness) {