Fixing softmax gradient bug.