Fix python 3.x tests