Add missing ensure