diff --git a/src/main/java/seng302/visualiser/controllers/dialogs/FinishDialogController.java b/src/main/java/seng302/visualiser/controllers/dialogs/FinishDialogController.java
index 34753497..2e22d074 100644
--- a/src/main/java/seng302/visualiser/controllers/dialogs/FinishDialogController.java
+++ b/src/main/java/seng302/visualiser/controllers/dialogs/FinishDialogController.java
@@ -5,6 +5,7 @@ import com.jfoenix.controls.JFXButton;
 import com.jfoenix.controls.JFXListView;
 import java.net.URL;
 import java.util.ArrayList;
+import java.util.Comparator;
 import java.util.ResourceBundle;
 import javafx.fxml.FXML;
 import javafx.fxml.Initializable;