Fix failing test due to version bump