Small update to the TODO