diff --git a/nms-patches/TileEntityEndGateway.patch b/nms-patches/TileEntityEndGateway.patch
index 2f1b789532..924a0849fa 100644
--- a/nms-patches/TileEntityEndGateway.patch
+++ b/nms-patches/TileEntityEndGateway.patch
@@ -13,6 +13,15 @@
  
  public class TileEntityEndGateway extends TileEntityEnderPortal implements ITickable {
  
+@@ -113,7 +119,7 @@
+     public void a(Entity entity) {
+         if (this.world instanceof WorldServer && !this.f()) {
+             this.c = 100;
+-            if (this.exitPortal == null && this.world.getDimensionKey() == World.THE_END) {
++            if (this.exitPortal == null && this.world.getTypeKey() == DimensionManager.THE_END) { // CraftBukkit - work in alternate worlds
+                 this.a((WorldServer) this.world);
+             }
+ 
 @@ -138,6 +144,26 @@
                      entity1 = entity.getRootVehicle();
                  }