Tras pasar tres años en la cárcel fue puesto en libertad y continuó dedicado en cuerpo y alma a la escritura de los "Shǐjì", que completó en el año 91 a. C., poco antes de morir.