Add a TODO comment indicating a desired refactoring