Fixing import issue with compat.PY26