Fix Python3 import