Fixed test to work on 2.6