From 593bebf930f88863a9fc763fc242828a40067d91 Mon Sep 17 00:00:00 2001
From: William Pederzoli <wpederzoli@gmail.com>
Date: Sun, 23 Jul 2023 03:01:45 +0200
Subject: [PATCH] gizmo plugin lag bugfix (#9166)

# Objective
Fixes #9156
---
 crates/bevy_gizmos/src/lib.rs | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/crates/bevy_gizmos/src/lib.rs b/crates/bevy_gizmos/src/lib.rs
index 3faf6c6305..34c1ce4262 100644
--- a/crates/bevy_gizmos/src/lib.rs
+++ b/crates/bevy_gizmos/src/lib.rs
@@ -18,7 +18,7 @@
 
 use std::mem;
 
-use bevy_app::{Last, Plugin, Update};
+use bevy_app::{Last, Plugin, PostUpdate};
 use bevy_asset::{load_internal_asset, AddAsset, Assets, Handle, HandleUntyped};
 use bevy_core::cast_slice;
 use bevy_ecs::{
@@ -50,7 +50,10 @@ use bevy_render::{
     view::RenderLayers,
     Extract, ExtractSchedule, Render, RenderApp, RenderSet,
 };
-use bevy_transform::components::{GlobalTransform, Transform};
+use bevy_transform::{
+    components::{GlobalTransform, Transform},
+    TransformSystem,
+};
 
 pub mod gizmos;
 
@@ -85,11 +88,12 @@ impl Plugin for GizmoPlugin {
             .init_resource::<GizmoStorage>()
             .add_systems(Last, update_gizmo_meshes)
             .add_systems(
-                Update,
+                PostUpdate,
                 (
                     draw_aabbs,
                     draw_all_aabbs.run_if(|config: Res<GizmoConfig>| config.aabb.draw_all),
-                ),
+                )
+                    .after(TransformSystem::TransformPropagate),
             );
 
         let Ok(render_app) = app.get_sub_app_mut(RenderApp) else { return; };