diff --git a/Spigot-Server-Patches/use-CB-BlockState-implementations-for-captured-block.patch b/Spigot-Server-Patches/use-CB-BlockState-implementations-for-captured-block.patch
index d1fee13149..485a2e2c43 100644
--- a/Spigot-Server-Patches/use-CB-BlockState-implementations-for-captured-block.patch
+++ b/Spigot-Server-Patches/use-CB-BlockState-implementations-for-captured-block.patch
@@ -18,7 +18,7 @@ the blockstate that will be valid for restoration, as opposed to dropping
 information on restoration when the event is cancelled.
 
 diff --git a/src/main/java/net/minecraft/server/World.java b/src/main/java/net/minecraft/server/World.java
-index 361cc6a92e..2bc0368d04 100644
+index 361cc6a92e..5a237fa1dd 100644
 --- a/src/main/java/net/minecraft/server/World.java
 +++ b/src/main/java/net/minecraft/server/World.java
 @@ -0,0 +0,0 @@ public abstract class World implements GeneratorAccess, AutoCloseable {
@@ -48,4 +48,13 @@ index 361cc6a92e..2bc0368d04 100644
                  this.capturedBlockStates.put(blockposition.immutableCopy(), blockstate);
              }
              // CraftBukkit end
+@@ -0,0 +0,0 @@ public abstract class World implements GeneratorAccess, AutoCloseable {
+     public IBlockData getType(BlockPosition blockposition) {
+         // CraftBukkit start - tree generation
+         if (captureTreeGeneration) {
+-            CapturedBlockState previous = capturedBlockStates.get(blockposition);
++            CraftBlockState previous = capturedBlockStates.get(blockposition); // Paper
+             if (previous != null) {
+                 return previous.getHandle();
+             }
 --
\ No newline at end of file