Fix import for python 3.2