fixed import for python2