diff --git a/src/lib.rs b/src/lib.rs
index 3471ec5..7f725cb 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -108,6 +108,7 @@ pub mod layout;
 pub mod overlay;
 pub mod style;
 pub mod theme;
+#[cfg(feature = "input-components")]
 pub(crate) mod undo;
 
 // Re-export commonly used types