correcred bug with badge display