finally,t hat elusive bug with softmax