diff --git a/src/main/java/org/bukkit/craftbukkit/util/BlockStateListPopulator.java b/src/main/java/org/bukkit/craftbukkit/util/BlockStateListPopulator.java
index c934af47f..9db620bd7 100644
--- a/src/main/java/org/bukkit/craftbukkit/util/BlockStateListPopulator.java
+++ b/src/main/java/org/bukkit/craftbukkit/util/BlockStateListPopulator.java
@@ -16,6 +16,7 @@ import net.minecraft.world.level.block.entity.TileEntity;
 import net.minecraft.world.level.block.state.IBlockData;
 import net.minecraft.world.level.dimension.DimensionManager;
 import net.minecraft.world.level.material.Fluid;
+import net.minecraft.world.level.storage.WorldData;
 import org.bukkit.block.BlockState;
 import org.bukkit.craftbukkit.block.CraftBlock;
 import org.bukkit.craftbukkit.block.CraftBlockEntityState;
@@ -141,4 +142,15 @@ public class BlockStateListPopulator extends DummyGeneratorAccess {
     public IRegistryCustom registryAccess() {
         return world.registryAccess();
     }
+
+    // Needed when a tree generates in water
+    @Override
+    public WorldData getLevelData() {
+        return world.getLevelData();
+    }
+
+    @Override
+    public long nextSubTickCount() {
+        return world.nextSubTickCount();
+    }
 }