Remove unneeded TODO comment

Closes #2