Import fix for python 3