fix softmax