Try to fix py3 imports.