This version works with Rspec test