Remove import which was causing a build failure for tests.