Put back the softmax doc.