2 files changed, 6 insertions, 4 deletions

diff --git a/src/ch/epfl/maze/tests/AnimalTest.java b/src/ch/epfl/maze/tests/AnimalTest.java
index 67c27e5..0b02442 100644
--- a/src/ch/epfl/maze/tests/AnimalTest.java
+++ b/src/ch/epfl/maze/tests/AnimalTest.java
@@ -6,10 +6,10 @@ import ch.epfl.maze.util.Vector2D;
 import junit.framework.TestCase;
 import org.junit.Test;
 
+import java.util.Set;
+
 /**
  * Test case for {@code Animal} implementation.
- *
- * @author EPFL
  */
 public class AnimalTest extends TestCase {
 
@@ -79,7 +79,7 @@ public class AnimalTest extends TestCase {
         }
 
         @Override
-        public Direction move(Direction[] choices) {
+        public Direction move(Set<Direction> choices) {
             return null;
         }
 
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;
             }