Add TP specific wrapper