Fix error `AttributeError: 'module' object has no attribute 'PY2'`