diff --git a/kls b/kls
index c5be34c..893cf58 100755
--- a/kls
+++ b/kls
@@ -424,7 +424,7 @@ async def main_async() -> None:
         await catch_input(selected_menu)
 
 
-def main() -> None:
+def main(screen: curses.window) -> None:
     asyncio.run(main_async())