BUG fix import for python 3