[IMP] Clean based on TDE comments