Fixing spec to reflect bug fix