Fix issue with rspec