Fix todo comment