diff --git a/core/src/test/java/art/arcane/iris/world/pregen/PregenPerformanceProfileTest.java b/core/src/test/java/art/arcane/iris/world/pregen/PregenPerformanceProfileTest.java index a2cff4331..78ab4b2d1 100644 --- a/core/src/test/java/art/arcane/iris/world/pregen/PregenPerformanceProfileTest.java +++ b/core/src/test/java/art/arcane/iris/world/pregen/PregenPerformanceProfileTest.java @@ -94,10 +94,15 @@ public class PregenPerformanceProfileTest { @Test public void raisingBurstParallelismOnlyGrowsThePool() { int before = MultiBurst.burst.parallelism(); - assertFalse(MultiBurst.burst.raiseParallelism(before)); - assertTrue(MultiBurst.burst.raiseParallelism(before + 1)); - assertEquals(before + 1, MultiBurst.burst.parallelism()); - assertFalse(MultiBurst.burst.raiseParallelism(before)); + try { + assertFalse(MultiBurst.burst.raiseParallelism(before)); + assertTrue(MultiBurst.burst.raiseParallelism(before + 1)); + assertEquals(before + 1, MultiBurst.burst.parallelism()); + assertFalse(MultiBurst.burst.raiseParallelism(before)); + } finally { + MultiBurst.burst.restoreParallelism(); + } + assertEquals(before, MultiBurst.burst.parallelism()); } @Test