fixed bug in softmax score