fix python 2 test