fixed import for python 3