Fix lobby assignment tests