Fix the end game