Fix a TODO reference