Fix so it will work with Rspec 2 and 3 both.