diff --git a/pyerrors/correlators.py b/pyerrors/correlators.py index 9f2cb197..95eadf74 100644 --- a/pyerrors/correlators.py +++ b/pyerrors/correlators.py @@ -916,7 +916,7 @@ class Corr: if self.N != 1: raise Exception("Correlator needs to be projected first.") - mc_names = list(set([item for sublist in [o[0].mc_names for o in self.content if o is not None] for item in sublist])) + mc_names = list(set([item for sublist in [sum(map(o[0].e_content.get, o[0].mc_names), []) for o in self.content if o is not None] for item in sublist])) x0_vals = [n for (n, o) in zip(np.arange(self.T), self.content) if o is not None] for name in mc_names: diff --git a/tests/correlators_test.py b/tests/correlators_test.py index c4980a7e..49332254 100644 --- a/tests/correlators_test.py +++ b/tests/correlators_test.py @@ -498,7 +498,8 @@ def test_corr_vector_operations(): def test_spaghetti_plot(): corr = _gen_corr(12, 50) - corr += pe.pseudo_Obs(0.0, 0.1, 'another_ensemble') + corr += pe.pseudo_Obs(0.0, 0.1, 'another_ensemble|r0') + corr += pe.pseudo_Obs(0.0, 0.1, 'another_ensemble|r1') corr += pe.cov_Obs(0.0, 0.01 ** 2, 'covobs') corr.spaghetti_plot(True)