'Changes the correct answer of the problem.'

def _change_correct_answer_for_problem(self, new_correct_choice=1):

    (unit, component) = self._get_problem_in_studio()
    modal = component.edit()
    modified_content = create_multiple_choice_xml(correct_choice=new_correct_choice)
    type_in_codemirror(self, 0, modified_content)
    modal.q(css='.action-save').click()
    unit.publish()
