Update/correct TODO