Updated the spec to match the new code