Fix bug and update TODOs.