Fix spec to match new output