Fixes import for python 3.4