Fix python3 import error