Small spec refactor