Fix relative import which will fail on Python 3.X