caja cambios automática del 406 2.0 hdi 2001

Hola buenos días. Tengo un problema con mi caja de cambios automática. Es la caja de cambios del peugeot 406 del año 2001 2.0 hdi. Creo que el modelo es la AL4. el problema viene cuando el coche alcanza la temperatura de 90º, en ese momento al parar el coche por ejemplo en un semáforo y reanudar la marcha pega un tirón muy molesto. Además noto cuando voy pisando el pedal casi parando el coche da un golpecito como si se frenára en seco o se saliera la marcha y luego al reanudar lo dicho golpe, en caliente. Tengo que decir que hemos cambiado el cuerpo de válvulas y lo sigue haciendo. Alguna ayuda? muchas gracias