From e7a129017e0891161d877bae4451eb72b8a80b21 Mon Sep 17 00:00:00 2001
From: Andy Park <chanhyup@andrew.cmu.edu>
Date: Fri, 8 Jul 2022 16:20:31 -0400
Subject: [PATCH] Fix #99

---
 jax_cosmo/angular_cl.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/jax_cosmo/angular_cl.py b/jax_cosmo/angular_cl.py
index 6db6c9f..afe5d4c 100644
--- a/jax_cosmo/angular_cl.py
+++ b/jax_cosmo/angular_cl.py
@@ -183,7 +183,6 @@ def gaussian_cl_covariance_and_mean(
     return_cls: (returns signal + noise cl, covariance)
     """
     ell = np.atleast_1d(ell)
-    n_ell = len(ell)
 
     # Compute signal vectors
     cl_signal = angular_cl(
@@ -191,7 +190,9 @@ def gaussian_cl_covariance_and_mean(
     )
     cl_noise = noise_cl(ell, probes)
 
+    return_cls = cl_signal + cl_noise
+
     # retrieve the covariance
     cov_mat = gaussian_cl_covariance(ell, probes, cl_signal, cl_noise, f_sky, sparse)
 
-    return cl_signal.flatten(), cov_mat
+    return return_cls.flatten(), cov_mat