Fix test for 2.6