TODO fixup