diff --git a/ppstructure/table/predict_structure.py b/ppstructure/table/predict_structure.py
index 7198fb2bcdc4d9d10f884f3a1545f23a1e628454..a580947aad428a0744e3da4b8302f047c6b11bee 100755
--- a/ppstructure/table/predict_structure.py
+++ b/ppstructure/table/predict_structure.py
@@ -148,7 +148,7 @@ def main(args):
                                                 bbox_list_str))
 
             if len(bbox_list) > 0 and len(bbox_list[0]) == 4:
-                img = draw_rectangle(image_file, pred_res['cell_bbox'])
+                img = draw_rectangle(image_file, bbox_list)
             else:
                 img = utility.draw_boxes(img, bbox_list)
             img_save_path = os.path.join(args.output,