Fixed compat import on python 3