improve rspec test