Remove TODO that I fixed :).