Update for deprecated dependency