small TODO fix