Remove this TODO