1 files changed, 6 insertions, 6 deletions
diff --git a/src/test/java/org/zwobble/couscous/tests/util/ExtraAsserts.java b/src/test/java/org/zwobble/couscous/tests/util/ExtraAsserts.java
index 98ea75a..c5f1668 100644
--- a/src/test/java/org/zwobble/couscous/tests/util/ExtraAsserts.java
+++ b/src/test/java/org/zwobble/couscous/tests/util/ExtraAsserts.java
@@ -1,19 +1,19 @@
 package org.zwobble.couscous.tests.util;
 
 import org.zwobble.couscous.util.Action;
+import org.zwobble.couscous.util.Casts;
+
+import lombok.SneakyThrows;
 
 public class ExtraAsserts {
-    @SuppressWarnings("unchecked")
+    @SneakyThrows
     public static <T extends Throwable> T assertThrows(Class<T> exceptionType, Action action) {
         try {
             action.run();
             throw new AssertionError("Expected exception");
         } catch (Exception exception) {
-            if (exceptionType.isInstance(exception)) {
-                return (T)exception;
-            } else {
-                throw exception;
-            }
+            return Casts.tryCast(exceptionType, exception)
+                .orElseThrow(() -> exception);
         }
     }
 }