update to account for api change