1 files changed, 3 insertions, 3 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;
         }