diff --git a/squirrelbattle/mapgeneration/broguelike.py b/squirrelbattle/mapgeneration/broguelike.py
index b261f57..ca830b8 100644
--- a/squirrelbattle/mapgeneration/broguelike.py
+++ b/squirrelbattle/mapgeneration/broguelike.py
@@ -71,8 +71,8 @@ class Generator:
         h, w = len(level), len(level[0])
         for y in range(h):
             for x in range(w):
-                if level[y][x] == Tile.FLOOR:
-                    for ny, nx in Map.neighbourhood(level, y, x):
+                if not level[y][x].is_wall():
+                    for ny, nx in Map.neighbourhood(level, y, x, large=True):
                         if level[ny][nx] == Tile.EMPTY:
                             level[ny][nx] = Tile.WALL