Update specs to match changes in #4