Adds check for assignment_mode weighted_random to finalize