Update to match new version