1 files changed, 10 insertions, 3 deletions
diff --git a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CustomItemEffects.java b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CustomItemEffects.java
index 72431697..93cbbf99 100644
--- a/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CustomItemEffects.java
+++ b/src/main/java/io/github/moulberry/notenoughupdates/miscfeatures/CustomItemEffects.java
@@ -96,6 +96,13 @@ public class CustomItemEffects {
 
 	private final Pattern etherwarpDistancePattern = Pattern.compile("up to (?<distance>\\d{2}) blocks away\\.");
 
+	private final List<String> aoteNames = Arrays.asList(
+		"ASPECT_OF_THE_END",
+		"ASPECT_OF_THE_VOID",
+		"ASPECT_OF_THE_LEECH_1",
+		"ASPECT_OF_THE_LEECH_2"
+	);
+
 	public int aoteTeleportationMillis = 0;
 	public Vector3f aoteTeleportationCurr = null;
 
@@ -196,7 +203,7 @@ public class CustomItemEffects {
 				if (tpTime <= 0 || Minecraft.getMinecraft().gameSettings.thirdPersonView != 0) return;
 
 				boolean aote = NotEnoughUpdates.INSTANCE.config.itemOverlays.enableSmoothAOTE &&
-					(internal.equals("ASPECT_OF_THE_END") || internal.equals("ASPECT_OF_THE_VOID"));
+					aoteNames.contains(internal);
 				boolean hyp = NotEnoughUpdates.INSTANCE.config.itemOverlays.enableSmoothHyperion && shadowWarp;
 				if (usingEtherwarp) {
 					tpTime = NotEnoughUpdates.INSTANCE.config.itemOverlays.smoothTpMillisEtherwarp;
@@ -759,7 +766,6 @@ public class CustomItemEffects {
 		add(Blocks.farmland);
 	}};
 
-
 	List<Block> scytheBlocks = Arrays.asList(
 		Blocks.leaves, Blocks.leaves2, Blocks.red_flower, Blocks.yellow_flower, Blocks.tallgrass, Blocks.double_plant);
 
@@ -1296,7 +1302,8 @@ public class CustomItemEffects {
 			HashSet<BlockPos> candidatesOld = new HashSet<>();
 			TreeMap<Float, Set<BlockPos>> candidatesOldSorted = new TreeMap<>();
 
-			getBuildersWandCandidates(Minecraft.getMinecraft().thePlayer,
+			getBuildersWandCandidates(
+				Minecraft.getMinecraft().thePlayer,
 				event.target,
 				event.partialTicks,
 				candidatesOld,