Fix test in 2.7