fix fireworks exit

2 files changed, 10 insertions, 5 deletions

diff --git a/internal/ui/fireworks.go b/internal/ui/fireworks.go
index c902a67..8442541 100644
--- a/internal/ui/fireworks.go
+++ b/internal/ui/fireworks.go
@@ -9,10 +9,11 @@ import (
 )
 
 type fwModel struct {
-	width  int
-	height int
-	start  time.Time
-	fws    []firework
+	width          int
+	height         int
+	start          time.Time
+	fws            []firework
+	ignoreFirstKey bool
 }
 
 type firework struct {
@@ -66,6 +67,10 @@ func (m fwModel) Update(msg tea.Msg) (tea.Model, tea.Cmd) {
 		}
 		return m, tick()
 	case tea.KeyMsg:
+		if !m.ignoreFirstKey {
+			m.ignoreFirstKey = true
+			return m, nil
+		}
 		return m, tea.Quit
 	}
 	return m, nil
diff --git a/internal/ui/table_test.go b/internal/ui/table_test.go
index 1161ec2..c2d37f5 100644
--- a/internal/ui/table_test.go
+++ b/internal/ui/table_test.go
@@ -202,7 +202,7 @@ func TestUndoHotkey(t *testing.T) {
 		os.Unsetenv("TASKRC")
 	})
 
-	m, err := New("")
+	m, err := New(nil)
 	if err != nil {
 		t.Fatalf("New: %v", err)
 	}