diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index 59f58b7..7b18483 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -268,7 +268,7 @@ def plot_predictions(filepath, input_items, colormap, model):
 			filepath,
 			[
 				# input_tensor,
-				input_items[1], #label_colourmap
+				input_pair[1], #label_colourmap
 				prediction_colormap
 			],
 			figsize=(18, 14)