Fix rspec bug