diff --git a/modules/amv/aeolus.py b/modules/amv/aeolus.py
index c71c753809620db28e9c8411b0852238e1c1707b..a2fccf4dde243dd74f7ab0f58b5ced90c3467865 100644
--- a/modules/amv/aeolus.py
+++ b/modules/amv/aeolus.py
@@ -116,15 +116,10 @@ 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 nn_idxs:
-            layers.append(a_profs[f_idx][k])
+        for k in range(len(on_earth)):
+            if on_earth[k]:
+                layers.append(a_profs[f_idx][k])
         a_profs[f_idx] = layers
 
         total += len(nn_idxs)