Fixed import issue in Py 3.