Add the exception handling logic for TC