Fix py3 issue in imports