Fix py3 import