Fix for rspec test