1 files changed, 4 insertions, 4 deletions

diff --git a/Drawer.cpp b/Drawer.cpp
index 6d5b228..8de3640 100644
--- a/Drawer.cpp
+++ b/Drawer.cpp
@@ -262,9 +262,9 @@ float Drawer::getImageWidth() const {
   
   for(Level::iterator object = window->getLevel().begin(); object != window->getLevel().end(); object++) {
     if((*object)->isOfType("Room")) {
-      Room *room = (Room*)&**object;
+      const Polygon &polygon = ((Room*)&**object)->getPolygon();
       
-      for(Room::iterator v = room->begin(); v != room->end(); v++) {
+      for(Polygon::const_iterator v = polygon.begin(); v != polygon.end(); v++) {
         max = fmaxf(max, fabsf(v->getX()));
       }
     }
@@ -278,9 +278,9 @@ float Drawer::getImageHeight() const {
   
   for(Level::iterator object = window->getLevel().begin(); object != window->getLevel().end(); object++) {
     if((*object)->isOfType("Room")) {
-      Room *room = (Room*)&**object;
+      const Polygon &polygon = ((Room*)&**object)->getPolygon();
       
-      for(Room::iterator v = room->begin(); v != room->end(); v++) {
+      for(Polygon::const_iterator v = polygon.begin(); v != polygon.end(); v++) {
         max = fmaxf(max, fabsf(v->getY()));
       }
     }