Removed incorrect imports from tests.