Fix spec to work correctly