Fix outdated TODO