Update to rspec tests