Minor - update TODO