Fix rspec tests