Removed the TODO comment that was done in the last commit.