diff --git a/modules/amv/aeolus.py b/modules/amv/aeolus.py
index 1506971c07262e3f8d476fc65a4d08a4e52d7ce6..c71c753809620db28e9c8411b0852238e1c1707b 100644
--- a/modules/amv/aeolus.py
+++ b/modules/amv/aeolus.py
@@ -116,10 +116,15 @@ def compare_aeolus_max_height(aeolus_dict, files_path, grid_value_name='cld_heig
         a_lats[f_idx] = a_lats[f_idx][on_earth]
         a_times[f_idx] = a_times[f_idx][on_earth]
 
+        # layers = []
+        # for k in range(len(nn_idxs)):
+        #     if on_earth[k]:
+        #         layers.append(a_profs[f_idx][k])
+        # a_profs[f_idx] = layers
+
         layers = []
-        for k in range(len(nn_idxs)):
-            if on_earth[k]:
-                layers.append(a_profs[f_idx][k])
+        for k in nn_idxs:
+            layers.append(a_profs[f_idx][k])
         a_profs[f_idx] = layers
 
         total += len(nn_idxs)