Fix tests with new extra argument for clone_or_update