Bugfix for Softmax.