diff --git a/create_mask_from_png.py b/create_mask_from_png.py
index 98a6779..5003a83 100644
--- a/create_mask_from_png.py
+++ b/create_mask_from_png.py
@@ -10,7 +10,7 @@ from config_file import UserConfigs
 # load the original input image and display it to our screen
 
 #filename = "equip/chests/chest_23_32"
-path = "equip/wands/"
+path = "flappy/"
 os.chdir(path)
 
 for entry in os.listdir():
diff --git a/litris.py b/litris.py
index 8f1acd7..f512846 100644
--- a/litris.py
+++ b/litris.py
@@ -73,7 +73,7 @@ class Litris(GameBase):
             if self.field.get_line_count() >= 6 and self.field.height() <= 2:
                 if self.field.predict_gaps_in_next_rotation() <= 5:
                     self.update_move_mode()
-                    self.field = self.stone_id_thread.get_current_board_state()
+                    self.field.state = self.stone_id_thread.get_current_board_state()
                     self.field.rotate_state()
                     #self.update_move_mode()
                     self.field.reset_half_field()