Fixed rspec issue