Forgot to modify the spec tests