Fix unit test to run locally as well as on Jenkins
Remove unused imports
Merge pull request #404 from joesondow/fix-unit-test-to-pass-locally

Fix unit test to run locally as well as on Jenkins