fix softmax cross entropy loss again