Attempted a fix on the lobby class