fix softmax mlp_backprop