update to reflect API changes