Improve the rspec test