fix the todo