Update for new version