Remove this TODO?