Fix PY2 compat