Tweak for rspec test passing