Fix spec to allow test suite to run