diff --git a/main.py b/main.py
index dcb4ebd..3a9e23f 100644
--- a/main.py
+++ b/main.py
@@ -159,7 +159,9 @@ class FlashcardCreator:
         # Create a scrollable column for the main content
         main_content = ft.Column(
             [
-                ft.Row([self.source_dropdown, self.target_dropdown, self.settings_button]),
+                ft.Row(
+                    [self.source_dropdown, self.target_dropdown, self.settings_button]
+                ),
                 ft.Divider(),
                 ft.Text("Input Text", size=16, weight=ft.FontWeight.BOLD),
                 self.input_text,
@@ -219,7 +221,7 @@ class FlashcardCreator:
                     f,
                 )
 
-            e.close(dialog)
+            e.page.close(dialog)
             e.page.update()
 
         # Load settings if they exist