Update for IO API changes