Remove helper function TODO