Update TODO
Fix compilation error