1 files changed, 3 insertions, 5 deletions
diff --git a/main.go b/main.go
index f173107..696a101 100644
--- a/main.go
+++ b/main.go
@@ -41,16 +41,14 @@ func main() {
 		log.Fatalf("Failed to create or open dictionary at '%s': %s", *dict, err)
 	}
 	defer db.Close()
-
-	err = setupDatabase(*rawDict, db)
-	if err != nil {
-		log.Fatalf("Failed to set up database: %s", err)
+	if err = setupTables(db); err != nil {
+		log.Fatalf("Failed to create database tables in dictionary '%s': %s", *dict, err)
 	}
 
 	c := http.DefaultClient
 	c.Timeout = 5 * time.Second
 
-	p := tea.NewProgram(initialModel(c, db, *apiURL, *deckName, *modelName, *initialWord))
+	p := tea.NewProgram(initialModel(c, db, *apiURL, *deckName, *modelName, *rawDict, *initialWord))
 	if _, err := p.Run(); err != nil {
 		log.Fatalf("Unexpected error encountered while running program: %s", err)
 	}