Fix a typo (I think) in Softmax class