1 files changed, 2 insertions, 3 deletions
diff --git a/cli/main.cpp b/cli/main.cpp
index c2d91ca..5172cd4 100644
--- a/cli/main.cpp
+++ b/cli/main.cpp
@@ -21,12 +21,11 @@ int main(int argc, const char **argv)
 
         auto screen = ftxui::ScreenInteractive::Fullscreen();
 
-        auto container = std::make_shared<katja::PageContainer>(std::vector<std::pair<std::string, katja::Page>>{
+        auto container = Screen(std::vector<std::pair<std::string, katja::Page>>{
             { "Home", std::make_shared<katja::WelcomePage>() },
             { "Updates", std::make_shared<katja::UpdatesPage>(std::move(updates)) },
             { "Search", std::make_shared<katja::SearchPage>() }
-        });
-        container->on_enter = screen.ExitLoopClosure();
+        }, screen.ExitLoopClosure());
 
         screen.Loop(container);
     }