fix python 3 compatibility in tests