fixing import for python3