Fix code comments to remove TAI and reflect new logic