fix imports for python2