Corrected imports in unit tests
Merge pull request #7 from cbracken/test-imports

Corrected imports in unit tests