Fix py3 incompat