Put back TP also for high PU scenarios