diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index f73819b..666e14f 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -166,7 +166,7 @@ else:
 		"metric_dice_coefficient": dice_coefficient,
 		"make_sensitivity": sensitivity,
 		"specificity": specificity,
-		"make_one_hot_mean_iou": mean_iou
+		"one_hot_mean_iou": mean_iou
 	})