Remove import which was causing a build failure for tests.
Merge pull request #14 from mrmonday/build-fix

Remove import which was causing a build failure for tests.