Fixing rspec tests