Remove incorrect import.
Merge pull request #90 from allenxwang/master

fix incorrect import