fix #3

plusls · Jun 22, 2021 · e9a2956 · e9a2956
1 parent 79ceaf1
commit e9a2956
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/src/main/java/com/plusls/ommc/feature/highlightLavaSource/LavaSourceResourceLoader.java b/src/main/java/com/plusls/ommc/feature/highlightLavaSource/LavaSourceResourceLoader.java
@@ -21,6 +21,7 @@
 import net.minecraft.util.Identifier;
 import net.minecraft.util.math.BlockPos;
 import net.minecraft.world.BlockRenderView;
+import org.jetbrains.annotations.Nullable;
 
 import java.util.function.Function;
 
@@ -63,10 +64,10 @@ public void reload(ResourceManager manager) {
         defaultLavaSourceSpites[1] = defaultLavaSourceFlowSprite;
         FluidRenderHandler lavaSourceRenderHandler = new FluidRenderHandler() {
             @Override
-            public Sprite[] getFluidSprites(BlockRenderView extendedBlockView, BlockPos blockPos, FluidState fluidState) {
+            public Sprite[] getFluidSprites(@Nullable BlockRenderView view, @Nullable BlockPos pos, FluidState state) {
 
-                if (Configs.FeatureToggle.HIGHLIGHT_LAVA_SOURCE.getBooleanValue()) {
-                    BlockState blockState = extendedBlockView.getBlockState(blockPos);
+                if (view != null && pos != null && Configs.FeatureToggle.HIGHLIGHT_LAVA_SOURCE.getBooleanValue()) {
+                    BlockState blockState = view.getBlockState(pos);
                     if (blockState.contains(FluidBlock.LEVEL) && blockState.get(FluidBlock.LEVEL) == 0) {
                         return lavaSourceSpites;
                     }