Fix bug. Fix spec expectation