fix tests broken in python3.x