fix bug due to API updates