Fix Rspec tests