Fix test class's 2.6- compatibility