diff --git a/studio/src/input.rs b/studio/src/input.rs
index 097c0a0..5991a93 100644
--- a/studio/src/input.rs
+++ b/studio/src/input.rs
@@ -41,7 +41,7 @@ pub enum Lifecycle {
 pub enum Action {
     Camera(CameraMotion),
     Selection(Selection),
-    ToggleDebug,
+    ToggleDebug
 }
 
 #[derive(Clone, Copy)]
diff --git a/studio/src/lib.rs b/studio/src/lib.rs
index ee3cc52..19291e5 100644
--- a/studio/src/lib.rs
+++ b/studio/src/lib.rs
@@ -157,6 +157,11 @@ pub fn start_viewer(
         },
     );
 
+    
+    // Only for registering components for undo
+    use space_editor::prelude::space_undo::AppAutoUndo;
+    app.auto_reflected_undo::<Transform>();
+
     #[cfg(feature = "wasm")]
     wasm::initialize();