fixed import for Python 3