No need to withdraw TPS. Keep TPS with EAD as backup. You can continue working under H1B cap. If lay-off happens and you do not enroll withing 240 days, you are out of status. That time TPS will save you. So, tell lawyer that TPS need not be withdrawn and it does not harm the H1B status.