diff --git a/pypeit/coadd2d.py b/pypeit/coadd2d.py
index ebe4689486..ac4a65082d 100644
--- a/pypeit/coadd2d.py
+++ b/pypeit/coadd2d.py
@@ -559,10 +559,8 @@ def coadd(self, interp_dspat=True):
             # if this is echelle data and the parset 'weights' is set to 'auto',
             # then the weights are computed per order, i.e., every order has a
             # different set of weights in each exposure (len(self.use_weights[slit_idx]) = nexp)
-            if self.pypeline == 'Echelle' and self.weights == 'auto':
-                _weights = self.use_weights[slit_idx]
-            else:
-                _weights = self.use_weights
+            _weights = self.use_weights[slit_idx] if self.pypeline == 'Echelle' and self.weights == 'auto' else self.use_weights
+            # TODO: Create a method here in the child clases? It is not great to do pypeline specific things in the parent
 
             # Perform the 2d coadd
             # NOTE: mask_stack is a gpm, and this is called inmask_stack in