diff --git a/paddlehub/serving/app_single.py b/paddlehub/serving/app_single.py
index 0b05af4a8f261c05ae8b63ed70ca42af281eaf4b..1cfeb94f83b148209a58c5b0020d0bab38eb4aff 100644
--- a/paddlehub/serving/app_single.py
+++ b/paddlehub/serving/app_single.py
@@ -147,7 +147,7 @@ def predict_mask(module, input_img, id, batch_size, extra=None, r_img=True):
             data.update(input_img)
         if extra is not None:
             data.update(extra)
-            r_img = False if "r_img" in extra.keys() else True
+            r_img = False if "visual_result" in extra.keys() else True
         results = predict_method(
             data=data,
             visualization=r_img,