diff --git a/src/display2d.rs b/src/display2d.rs
index 2cdce72..697dc32 100644
--- a/src/display2d.rs
+++ b/src/display2d.rs
@@ -35,7 +35,7 @@ impl Plugin for Display2dPlugin {
                     set_transform
                         .after(game::update_board)
                         .run_if(any_component_changed::<BoardIndex>),
-                    set_piece_sprite.run_if(any_component_changed::<Side>),
+                    set_piece_sprite.run_if(any_component_changed::<Side>).after(game::set_side),
                     set_tile_sprite.run_if(any_component_added::<game::Tile>),
                 ),
             )
diff --git a/src/display3d.rs b/src/display3d.rs
index f130f17..0480f16 100644
--- a/src/display3d.rs
+++ b/src/display3d.rs
@@ -35,7 +35,7 @@ impl Plugin for Display3dPlugin {
                     set_board_model.run_if(any_component_added::<game::BoardComponent>),
                     set_tile_hitbox.run_if(any_component_added::<game::Tile>),
                     set_piece_position.run_if(any_component_changed::<BoardIndex>),
-                    set_piece_texture.run_if(any_component_changed::<Side>),
+                    set_piece_texture.run_if(any_component_changed::<Side>).after(game::set_side),
                     select
                         .run_if(in_state(GameState::Play))
                         .run_if(in_state(DisplayState::Display3d))
diff --git a/src/game.rs b/src/game.rs
index e42c506..c3da7e7 100644
--- a/src/game.rs
+++ b/src/game.rs
@@ -30,13 +30,13 @@ impl Plugin for GamePlugin {
                         .run_if(any_component_added::<Selected>),
                 ),
             )
-            // .add_systems(
-            //     PreUpdate,
-            //     (
-            //         asserts::<display2d::Display2d>,
-            //         asserts::<display3d::Display3d>,
-            //     ),
-            // )
+            .add_systems(
+                PreUpdate,
+                (
+                    asserts::<display2d::Display2d>,
+                    asserts::<display3d::Display3d>,
+                ),
+            )
             .add_systems(
                 PostUpdate,
                 (debug_board.run_if(resource_exists::<debug::DebugEnabled>()),),