Fix local imports for py3 compatibility.