Fix ImportError in Python3