diff --git a/src/tui/app/machine/mod.rs b/src/tui/app/machine/mod.rs
index 0295e16..5aee12b 100644
--- a/src/tui/app/machine/mod.rs
+++ b/src/tui/app/machine/mod.rs
@@ -26,7 +26,7 @@ use match_state::MatchState;
 use reload_state::ReloadState;
 use search_state::SearchState;
 
-use super::{AppMode, IAppBase};
+use super::{AppMode, IAppBase, IAppState};
 
 pub type App = AppState<
     AppMachine<BrowseState>,
@@ -130,36 +130,11 @@ impl IApp for App {
         self
     }
 
-    fn state(
-        self,
-    ) -> AppState<
-        Self::BrowseState,
-        Self::InfoState,
-        Self::ReloadState,
-        Self::SearchState,
-        Self::FetchState,
-        Self::MatchState,
-        Self::ErrorState,
-        Self::CriticalState,
-    > {
+    fn state(self) -> IAppState!() {
         self
     }
 
-    fn mode(
-        mut self,
-    ) -> super::AppMode<
-        AppState<
-            Self::BrowseState,
-            Self::InfoState,
-            Self::ReloadState,
-            Self::SearchState,
-            Self::FetchState,
-            Self::MatchState,
-            Self::ErrorState,
-            Self::CriticalState,
-        >,
-        Self::InputMode,
-    > {
+    fn mode(mut self) -> super::AppMode<IAppState!(), Self::InputMode> {
         if let Some(input) = self.inner_mut().input.take() {
             AppMode::Input(AppInputMode::new(input, self.state()))
         } else {
diff --git a/src/tui/app/mod.rs b/src/tui/app/mod.rs
index 177860b..644f7f7 100644
--- a/src/tui/app/mod.rs
+++ b/src/tui/app/mod.rs
@@ -30,6 +30,7 @@ macro_rules! IAppState {
                  Self::FetchState, Self::MatchState, Self::ErrorState, Self::CriticalState>
     };
 }
+use IAppState;
 
 pub trait IApp {
     type BrowseState: IAppBase<APP = Self> + IAppInteractBrowse<APP = Self>;