fix import - remove unused code