Fixed an import issue to support py26.