diff --git a/modules/aeolus/aeolus_amv.py b/modules/aeolus/aeolus_amv.py
index 0c266deb291308ebf39355fff586ddd2b6d3c090..c3c6c2647e484578905f2c41521c78818fbd6f08 100644
--- a/modules/aeolus/aeolus_amv.py
+++ b/modules/aeolus/aeolus_amv.py
@@ -664,8 +664,8 @@ def subset_by_time(match_file, dt_str_0, dt_str_1, amv_var_names):
     coords = {'num_profs': times[time_idxs], 'num_params': amv_var_names}
     amvs_da = xr.DataArray(amvs, coords=coords, dims=['num_profs', 'max_num_amvs', 'num_params'])
 
-    prof_locs_da = xr.DataArray(np.column_stack([lons[time_idxs], lats[time_idxs]]),
-                                coords=[times[time_idxs], ['longitude', 'latitude']],
+    prof_locs_da = xr.DataArray(np.column_stack([lons[time_idxs], lats[time_idxs], n_amvs_per_prof[time_idxs], n_levs_per_prof[time_idxs]]),
+                                coords=[times[time_idxs], ['longitude', 'latitude', 'num_amvs', 'num_levels']],
                                 dims=['num_profs', 'space'])
 
     return prof_da, amvs_da, prof_locs_da