Remove unnecessary imports in tests