fix unit test imports
Merge pull request #2 from oberhamsi/master

fix unit tests