Fix for rspec