diff --git a/src/test/java/org/highmed/service/executors/CohortExecutorTest.java b/src/test/java/org/highmed/service/executors/CohortExecutorTest.java
index 547f5fe5..c0b08bc5 100644
--- a/src/test/java/org/highmed/service/executors/CohortExecutorTest.java
+++ b/src/test/java/org/highmed/service/executors/CohortExecutorTest.java
@@ -9,6 +9,7 @@
 import org.mockito.junit.MockitoJUnitRunner;
 import org.highmed.domain.model.*;
 import org.highmed.service.ehrbase.EhrBaseService;
+import org.highmed.service.exception.IllegalArgumentException;
 
 import java.util.List;
 import java.util.Map;
diff --git a/src/test/java/org/highmed/service/executors/SetOperationsServiceTest.java b/src/test/java/org/highmed/service/executors/SetOperationsServiceTest.java
index 91aa88f6..0f5279ff 100644
--- a/src/test/java/org/highmed/service/executors/SetOperationsServiceTest.java
+++ b/src/test/java/org/highmed/service/executors/SetOperationsServiceTest.java
@@ -6,6 +6,7 @@
 import org.mockito.InjectMocks;
 import org.mockito.junit.MockitoJUnitRunner;
 import org.highmed.domain.model.Operator;
+import org.highmed.service.exception.IllegalArgumentException;
 
 import java.util.Collections;
 import java.util.LinkedList;