Fix test failure caused by version bump