When the team is separated during a riot at a mental institution, Mallory poses as a doctor to rescue Diana, who's been mistaken for a patient.