Missed to update all files after API change