My suggestion is to show the play again or the thank you for playing after you die, and not in the end of the match.
Because not of only that u have to wait for end to press play again, when the play again show up the match already ended so you don't have time to press on it before it transfer you back to the lobby.
Because not of only that u have to wait for end to press play again, when the play again show up the match already ended so you don't have time to press on it before it transfer you back to the lobby.