Add test to spec.rb