Fixing todo comment