Fixed todo comment