diff --git a/tui/screens/main_screen.py b/tui/screens/main_screen.py
index 28e3e22..ad8fa3e 100644
--- a/tui/screens/main_screen.py
+++ b/tui/screens/main_screen.py
@@ -7,7 +7,7 @@ from typing import TYPE_CHECKING, Any
 from textual.app import ComposeResult
 from textual.binding import Binding
 from textual.containers import Horizontal, Vertical
-from textual.screen import Screen
+from textual.screen import ModalScreen, Screen
 from textual.widgets import Button, DataTable, Footer, Header, Input, RichLog, Static
 from textual import work