refactor: cleanup import