Refactor: remove unnecessary import