Fix python 3 compatibility issue in tests