Fix rspec test