Minor fix, update todo.