Fix testing for 2.6