fix a TODO