Improve last todo-mode fix