diff --git a/src/Automaton.hs b/src/Automaton.hs index a3c45fc..98ea890 100644 --- a/src/Automaton.hs +++ b/src/Automaton.hs @@ -70,11 +70,11 @@ answer (Message.Relay {Message.from, Message.message = Message.LogIn {Message.na | from == key = return $ LoggedIn {key, name} answer (Message.Relay {Message.from, Message.message = Message.Invitation {}}) (LoggedIn {key, name}) = do - send $ Message.Answer {Message.accept = True} + send $ Message.Answer {Message.accept = True, Message.to = from} return $ Playing {key, name, against = from} -answer (Message.Relay {Message.message = Message.Invitation {}}) state = do - send $ Message.Answer {Message.accept = False} +answer (Message.Relay {Message.from, Message.message = Message.Invitation {}}) state = do + send $ Message.Answer {Message.accept = False, Message.to = from} return state answer message@(Message.Game {Message.logs}) state@(Playing {key, name}) = do