diff --git a/modules/deeplearning/icing_fcn.py b/modules/deeplearning/icing_fcn.py
index 358b29b3cdd1ecb3fdc1eec55e8faf5c6a02fc68..c7c305f29a64c7a20df60423cd2a5da04444931d 100644
--- a/modules/deeplearning/icing_fcn.py
+++ b/modules/deeplearning/icing_fcn.py
@@ -555,6 +555,11 @@ class IcingIntensityFCN:
         flt_alt = h5f['flight_altitude'][:]
         tst_idxs = np.arange(time.shape[0])
 
+        # For testing, make better
+        # print('total: ', len(tst_idxs))
+        # keep = flt_alt > 3000
+        # tst_idxs = tst_idxs[keep]
+
         self.num_data_samples = len(tst_idxs)
 
         self.get_test_dataset(tst_idxs)