Fix rspec problem too