Modify rspec tests