softmax fix.