diff --git a/src/main/java/p2/wordsuggestor/NGramToNextChoicesMap.java b/src/main/java/p2/wordsuggestor/NGramToNextChoicesMap.java
index 7f66f8433375cce24d2c4edd2a9b6f9502bee5da..111d10294dc8e93e1863f0860720162f6b477a85 100644
--- a/src/main/java/p2/wordsuggestor/NGramToNextChoicesMap.java
+++ b/src/main/java/p2/wordsuggestor/NGramToNextChoicesMap.java
@@ -7,6 +7,8 @@ import cse332.misc.LargeValueFirstItemComparator;
 import cse332.sorts.InsertionSort;
 import cse332.types.AlphabeticString;
 import cse332.types.NGram;
+import p2.sorts.QuickSort;
+import p2.sorts.TopKSort;
 
 import java.util.Comparator;
 import java.util.Iterator;