fix softmax cross entropy