Fix rspec error