diff --git a/easyunfold/effective_mass.py b/easyunfold/effective_mass.py
index e0d416a..e1162b2 100644
--- a/easyunfold/effective_mass.py
+++ b/easyunfold/effective_mass.py
@@ -144,8 +144,8 @@ def get_band_extrema(self, mode: str = 'cbm', extrema_tol: float = None, ispin=0
         cbm_indices = []
         for ik, wset in enumerate(weights):
             for isubset in range(wset.shape[1]):
-                etmp = weights[ik][ispin, isubset, :, 0]
-                wtmp = weights[ik][ispin, isubset, :, 1]
+                etmp = wset[ispin, isubset, :, 0]
+                wtmp = wset[ispin, isubset, :, 1]
                 # Filter by intensity
                 mask = wtmp > intensity_tol
                 midx = np.where(mask)[0]