Fix YUM UI, where mock did not return task instance