1 files changed, 3 insertions, 1 deletions

diff --git a/src/ch/epfl/maze/tests/GhostsTest.java b/src/ch/epfl/maze/tests/GhostsTest.java
index 2195443..c6970b1 100644
--- a/src/ch/epfl/maze/tests/GhostsTest.java
+++ b/src/ch/epfl/maze/tests/GhostsTest.java
@@ -16,6 +16,8 @@ import ch.epfl.maze.util.Vector2D;
 import junit.framework.TestCase;
 import org.junit.Test;
 
+import java.util.Set;
+
 /**
  * Test suite for ghosts implementation.
  *
@@ -126,7 +128,7 @@ public class GhostsTest extends TestCase {
         }
 
         @Override
-        public Direction move(Direction[] choices, Daedalus daedalus) {
+        public Direction move(Set<Direction> choices, Daedalus daedalus) {
             if (mMoves) {
                 return getPosition().getX() % 2 == 0 ? Direction.RIGHT : Direction.LEFT;
             }