fix softmax cross entropy loss