diff --git a/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java b/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
index ff587dcb8f2b516689eb66dd072eed176d2eb5f9..2a7838299662dca2ba029650d749a06d27b01a69 100644
--- a/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
+++ b/src/tests/gitlab/ckpt1/NGramToNextChoicesMapTests.java
@@ -10,16 +10,16 @@ import cse332.datastructures.containers.Item;
 import cse332.interfaces.misc.Dictionary;
 import cse332.types.AlphabeticString;
 import cse332.types.NGram;
-import datastructures.dictionaries.HashTrieMap;
+import cse332.datastructures.trees.BinarySearchTree;
 import p2.wordsuggestor.NGramToNextChoicesMap;
 import tests.TestsUtility;
 
 public class NGramToNextChoicesMapTests extends TestsUtility {
     private static Supplier<Dictionary<NGram, Dictionary<AlphabeticString, Integer>>> newOuter =
-            () -> new HashTrieMap<String, NGram, Dictionary<AlphabeticString, Integer>>(NGram.class);
+            () -> new BinarySearchTree();
 
-    private static Supplier<Dictionary<AlphabeticString, Integer>> newInner = 
-            () -> new HashTrieMap<Character, AlphabeticString, Integer>(AlphabeticString.class);
+    private static Supplier<Dictionary<AlphabeticString, Integer>> newInner =
+            () -> new BinarySearchTree();
     
     public static void main(String[] args) {
         new NGramToNextChoicesMapTests().run();