TODO patch

Only a small TODO update