Fix specs due to ensure changes