Allow for set_expected_response