diff --git a/squirrelbattle/tests/mapgeneration_test.py b/squirrelbattle/tests/mapgeneration_test.py
index ed0951c..4fc6b28 100644
--- a/squirrelbattle/tests/mapgeneration_test.py
+++ b/squirrelbattle/tests/mapgeneration_test.py
@@ -23,7 +23,7 @@ class TestRandomWalk(unittest.TestCase):
         self.assertTrue(m.tiles[m.start_y][m.start_x].can_walk())
 
         #DFS
-        queue = m.neighbourhood(m.start_y, m.start_x)
+        queue = Map.neighbourhood(m.tiles, m.start_y, m.start_x)
         while queue != []:
             y, x = queue.pop()
             if m.tiles[y][x].can_walk():