fix unittest for 2.6