updated to reflect api change