Fix python 3.x test