diff --git a/src/test/java/seng302/models/BoatGroupTest.java b/src/test/java/seng302/models/BoatGroupTest.java
index 441a0429..60c18cc7 100644
--- a/src/test/java/seng302/models/BoatGroupTest.java
+++ b/src/test/java/seng302/models/BoatGroupTest.java
@@ -1,5 +1,5 @@
 package seng302.models;
-
+import seng302.*;
 import javafx.scene.paint.*;
 import javafx.scene.paint.Color;
 import javafx.scene.shape.Polygon;
diff --git a/src/test/java/seng302/models/MarkGroupTest.java b/src/test/java/seng302/models/MarkGroupTest.java
index 4a0de560..994bc616 100644
--- a/src/test/java/seng302/models/MarkGroupTest.java
+++ b/src/test/java/seng302/models/MarkGroupTest.java
@@ -1,5 +1,6 @@
 package seng302.models;
 
+import seng302.*;
 import javafx.geometry.Point2D;
 import org.junit.Assert;
 import org.junit.Before;