fix ImportError
Merge pull request #173 from ChenShuang/fix-bugs

fix ImportError