diff --git a/modules/deeplearning/srcnn_l1b_l2.py b/modules/deeplearning/srcnn_l1b_l2.py
index 859c4942a8abe49583dedcb331799194018bc965..eae290fb36cf4924140d1176978e22ec053726e5 100644
--- a/modules/deeplearning/srcnn_l1b_l2.py
+++ b/modules/deeplearning/srcnn_l1b_l2.py
@@ -54,8 +54,8 @@ label_params = ['cloud_fraction']
 
 DO_ZERO_OUT = False
 
-label_idx = 0
-label_param = label_params[label_idx]
+label_idx = 3
+label_param = params[label_idx]
 print('data_params: ', data_params)
 print('label_params: ', label_params)