Add a TODO: remove redundancy