Fix import path for python 3.