diff --git a/contact/ui/dialog.py b/contact/ui/dialog.py
index 3dd09ce..42823d3 100644
--- a/contact/ui/dialog.py
+++ b/contact/ui/dialog.py
@@ -3,7 +3,9 @@ from contact.ui.colors import get_color
 
 
 def dialog(title: str, message: str) -> None:
+    """Display a dialog with a title and message."""
 
+    curses.update_lines_cols()
     height, width = curses.LINES, curses.COLS
 
     # Calculate dialog dimensions
diff --git a/contact/ui/ui_state.py b/contact/ui/ui_state.py
index ce5d643..80f9278 100644
--- a/contact/ui/ui_state.py
+++ b/contact/ui/ui_state.py
@@ -24,12 +24,12 @@ class ChatUIState:
     selected_message: int = 0
     selected_node: int = 0
     current_window: int = 0
+    last_sent_time: float = 0.0
 
     selected_index: int = 0
     start_index: List[int] = field(default_factory=lambda: [0, 0, 0])
     show_save_option: bool = False
     menu_path: List[str] = field(default_factory=list)
-    last_sent_time: float = 0.0
 
 
 @dataclass