diff --git a/src/GomokuAI.java b/src/GomokuAI.java
index 4dace93..a583422 100644
--- a/src/GomokuAI.java
+++ b/src/GomokuAI.java
@@ -1,7 +1,7 @@
 import java.util.List;
 import java.util.Map;
 import java.util.Random;
-import java.util.List;
+import java.util.HashMap;
 
 public class GomokuAI extends Player {