Fix a tiny bug, clean-up TODOs.