Fix the todo function