1 files changed, 2 insertions, 2 deletions
diff --git a/public/assets/javascripts/rectangles/engine/scenery/undo.js b/public/assets/javascripts/rectangles/engine/scenery/undo.js
index 54ab755..a7e7d61 100644
--- a/public/assets/javascripts/rectangles/engine/scenery/undo.js
+++ b/public/assets/javascripts/rectangles/engine/scenery/undo.js
@@ -20,7 +20,7 @@
       undo: function(state){
         var scenery = Scenery.find(state.id)
         scenery.deserialize(state)
-        scenery.set_wall(Rooms.walls[ state.wall_id ])
+        scenery.set_wall(Walls.find( state.wall_id ))
         
         if (editor.permissions.resize) {
           Scenery.resize.show(scenery)
@@ -32,7 +32,7 @@
       redo: function(state){
         var scenery = Scenery.find(state.id)
         scenery.deserialize(state)
-        scenery.set_wall(Rooms.walls[ state.wall_id ])
+        scenery.set_wall(Walls.find( state.wall_id ))
 
         if (editor.permissions.resize) {
           Scenery.resize.show(scenery)