Fix for end game