Fix a TODO