Fix to RSpec test