Export marginals instead of probs