Fix RSpec tests