diff --git a/main.go b/main.go
index 9fbe5e9..cce0386 100644
--- a/main.go
+++ b/main.go
@@ -35,6 +35,8 @@ func main() {
 
 	// Setup state machine.
 	fsm := machine.New()
+	fsm.Register("menu", &handlers.Menu{})
+	fsm.Register("gameover", &handlers.GameOver{})
 	fsm.Register("gameplay", handlers.NewGamePlay())
 	fsm.Start("gameplay")