diff --git a/src/test/java/oving7/savingsaccount/ForeldreSparTest.java b/src/test/java/oving7/savingsaccount/ForeldreSparTest.java
index 11f6ec5..046f20b 100644
--- a/src/test/java/oving7/savingsaccount/ForeldreSparTest.java
+++ b/src/test/java/oving7/savingsaccount/ForeldreSparTest.java
@@ -31,7 +31,7 @@ public class ForeldreSparTest {
 
 		assertEquals(9000, foreldreSpar.getBalance(), epsilon, "The account balance is incorrect");
 
-		assertThrows(IllegalArgumentException.class, () -> {
+		assertThrows(IllegalStateException.class, () -> {
 			foreldreSpar.withdraw(10_000);
 		}, "Should not be able to withdraw more money than is in the account");
 
diff --git a/src/test/java/oving7/savingsaccount/SavingsAccountTest.java b/src/test/java/oving7/savingsaccount/SavingsAccountTest.java
index 88710e2..704ff53 100644
--- a/src/test/java/oving7/savingsaccount/SavingsAccountTest.java
+++ b/src/test/java/oving7/savingsaccount/SavingsAccountTest.java
@@ -57,9 +57,9 @@ public class SavingsAccountTest {
 
 		savingsAccount.deposit(10.0);
 
-		assertThrows(IllegalArgumentException.class, () -> {
+		assertThrows(IllegalStateException.class, () -> {
 			savingsAccount.withdraw(40.0);
-		}, "Withdrawal of more than the balance should have triggered an IllegalArgumentException");
+		}, "Withdrawal of more than the balance should have triggered an IllegalStateException");
 
 		assertEquals(30.0, savingsAccount.getBalance(), epsilon,
 				"The account balance was incorrect");