Fix old import

Was still referring to the old repository.
Merge pull request #2 from wongak/master

Fix old import path