Add TODO: move to Utils