Fix import package
Merge pull request #13 from matthughes/master

Fix import package