Change imports to be absolute on Python 2 as well