Fix spec to make test suite pass