Fixed the todo