Minor TODO fixup