Fixes a quick bug with the lobby code