diff --git a/src/main/java/net/wurstclient/test/PistonTest.java b/src/main/java/net/wurstclient/test/PistonTest.java
index 83c90c486b..387548549d 100644
--- a/src/main/java/net/wurstclient/test/PistonTest.java
+++ b/src/main/java/net/wurstclient/test/PistonTest.java
@@ -9,8 +9,6 @@
 
 import static net.wurstclient.test.WurstClientTestHelper.*;
 
-import java.time.Duration;
-
 import net.minecraft.block.Blocks;
 
 public enum PistonTest
@@ -27,13 +25,13 @@ public static void testPistonDoesntCrash()
 		waitForBlock(0, 1, 2, Blocks.PISTON);
 		runChatCommand("setblock ~ ~ ~2 minecraft:redstone_block");
 		waitForBlock(0, 0, 2, Blocks.REDSTONE_BLOCK);
-		takeScreenshot("piston_extending", Duration.ZERO);
+		takeScreenshot("piston_extending");
 		waitForWorldTicks(3);
 		
 		// Destroy the redstone block
 		runChatCommand("setblock ~ ~ ~2 minecraft:air");
 		waitForBlock(0, 0, 2, Blocks.AIR);
-		takeScreenshot("piston_retracting", Duration.ZERO);
+		takeScreenshot("piston_retracting");
 		waitForWorldTicks(3);
 		
 		// Clean up
diff --git a/src/main/java/net/wurstclient/test/XRayHackTest.java b/src/main/java/net/wurstclient/test/XRayHackTest.java
index 88da924161..e8bdbd295c 100644
--- a/src/main/java/net/wurstclient/test/XRayHackTest.java
+++ b/src/main/java/net/wurstclient/test/XRayHackTest.java
@@ -25,7 +25,7 @@ public static void testXRayHack()
 		runWurstCommand("setcheckbox X-Ray only_show_exposed off");
 		runWurstCommand("setslider X-Ray opacity 0");
 		runWurstCommand("t X-Ray on");
-		takeScreenshot("xray_default", Duration.ofMillis(250));
+		takeScreenshot("xray_default", Duration.ofMillis(300));
 		runWurstCommand("t X-Ray off");
 		clearChat();
 		
@@ -33,7 +33,7 @@ public static void testXRayHack()
 		runWurstCommand("setcheckbox X-Ray only_show_exposed on");
 		runWurstCommand("setslider X-Ray opacity 0");
 		runWurstCommand("t X-Ray on");
-		takeScreenshot("xray_exposed_only", Duration.ofMillis(250));
+		takeScreenshot("xray_exposed_only", Duration.ofMillis(300));
 		runWurstCommand("t X-Ray off");
 		clearChat();
 		
@@ -41,7 +41,7 @@ public static void testXRayHack()
 		runWurstCommand("setcheckbox X-Ray only_show_exposed off");
 		runWurstCommand("setslider X-Ray opacity 0.5");
 		runWurstCommand("t X-Ray on");
-		takeScreenshot("xray_opacity", Duration.ofMillis(250));
+		takeScreenshot("xray_opacity", Duration.ofMillis(300));
 		runWurstCommand("t X-Ray off");
 		clearChat();
 		
@@ -49,7 +49,7 @@ public static void testXRayHack()
 		runWurstCommand("setcheckbox X-Ray only_show_exposed on");
 		runWurstCommand("setslider X-Ray opacity 0.5");
 		runWurstCommand("t X-Ray on");
-		takeScreenshot("xray_exposed_only_opacity", Duration.ofMillis(250));
+		takeScreenshot("xray_exposed_only_opacity", Duration.ofMillis(300));
 		runWurstCommand("t X-Ray off");
 		clearChat();