From 6bc9d6de75eaf3af16e8545524f4e74d669b113c Mon Sep 17 00:00:00 2001
From: Timo Ley <tilera.mpz@gmail.com>
Date: Sat, 18 May 2019 21:15:24 +0200
Subject: [PATCH] Snap now works in all Dimensions

---
 src/main/java/anvil/infinity/snap/SnapHelper.java | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/src/main/java/anvil/infinity/snap/SnapHelper.java b/src/main/java/anvil/infinity/snap/SnapHelper.java
index 3ec5d2e..ac8a368 100644
--- a/src/main/java/anvil/infinity/snap/SnapHelper.java
+++ b/src/main/java/anvil/infinity/snap/SnapHelper.java
@@ -20,8 +20,10 @@ import net.minecraft.util.text.TextComponentTranslation;
 import net.minecraft.util.text.TextFormatting;
 import net.minecraft.world.GameType;
 import net.minecraft.world.World;
+import net.minecraft.world.WorldServer;
 import net.minecraftforge.fml.common.FMLCommonHandler;
 
+import java.util.ArrayList;
 import java.util.List;
 import java.util.Random;
 
@@ -32,9 +34,12 @@ public class SnapHelper {
     public static boolean snap(EntityLivingBase entity) {
         EntityData data = GauntletUserInformation.getDataByEntity(entity);
         if (GauntelHelper.hasFullGauntlet(entity)) {
-            World w = entity.getEntityWorld();
+            WorldServer[] worlds = server.worlds;
             if (data.selectedSnapResult == SnapResult.KILLHALF) {
-                List<Entity> entities = w.loadedEntityList;
+                List<Entity> entities = new ArrayList<Entity>();
+                for (int i = 0; i < worlds.length; i++) {
+                    entities.addAll(worlds[i].loadedEntityList);
+                }
                 boolean kill = false;
 
                 PlayerList players = server.getPlayerList();