minor update on TODO