diff --git a/pkg/ui/load_save_screens.go b/pkg/ui/load_save_screens.go
index 922cfa5..bd6d0f0 100644
--- a/pkg/ui/load_save_screens.go
+++ b/pkg/ui/load_save_screens.go
@@ -107,8 +107,7 @@ func (se *SchemaEditor) showLoadScreen() {
 
 	// Keyboard shortcuts
 	form.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
-		switch event.Key() {
-		case tcell.KeyEscape:
+		if event.Key() == tcell.KeyEscape {
 			se.app.Stop()
 			return nil
 		}
@@ -214,8 +213,7 @@ func (se *SchemaEditor) showSaveScreen() {
 
 	// Keyboard shortcuts
 	form.SetInputCapture(func(event *tcell.EventKey) *tcell.EventKey {
-		switch event.Key() {
-		case tcell.KeyEscape:
+		if event.Key() == tcell.KeyEscape {
 			se.pages.RemovePage("save-database")
 			se.pages.SwitchToPage("main")
 			return nil