Updated spec to work with #2 fix