--- a/nagslang/level.py	Sun Sep 01 18:34:38 2013 +0200
+++ b/nagslang/level.py	Sun Sep 01 18:39:37 2013 +0200
@@ -4,13 +4,13 @@
 from nagslang.resources import resources
 
 POLY_COLORS = {
-        1: pygame.color.THECOLORS['red'],
-        2: pygame.color.THECOLORS['green'],
-        3: pygame.color.THECOLORS['yellow'],
-        4: pygame.color.THECOLORS['blue'],
-        5: pygame.color.THECOLORS['lightblue'],
-        6: pygame.color.THECOLORS['magenta']
-        }
+    1: pygame.color.THECOLORS['red'],
+    2: pygame.color.THECOLORS['green'],
+    3: pygame.color.THECOLORS['yellow'],
+    4: pygame.color.THECOLORS['blue'],
+    5: pygame.color.THECOLORS['lightblue'],
+    6: pygame.color.THECOLORS['magenta'],
+}
 
 
 class Level(object):
@@ -76,7 +76,7 @@
         return (pos[0], self.y - pos[1])
 
     def get_walls(self):
-        return self.polygons[1]
+        return self.polygons.values()
 
     def _draw_walls(self):
         for index, polygon in self.polygons.items():