diff --git a/examples/tm_mobilenet_ssd_uint8_timvx.cpp b/examples/tm_mobilenet_ssd_uint8_timvx.cpp
index ec86b4dc44a3528e370fb8260d0c03fe1fc82253..14a1888f522004c0217368876c8f1655eabc64ca 100644
--- a/examples/tm_mobilenet_ssd_uint8_timvx.cpp
+++ b/examples/tm_mobilenet_ssd_uint8_timvx.cpp
@@ -185,9 +185,9 @@ int main(int argc, char* argv[])
 
     // create VeriSilicon TIM-VX backend
     context_t timvx_context = create_context("timvx", 1);
-    if ( set_context_device(timvx_context, "TIMVX", nullptr, 0) < 0 )
+    if (set_context_device(timvx_context, "TIMVX", nullptr, 0) < 0)
     {
-        fprintf(stderr, "add_context_device failed.\n" );
+        fprintf(stderr, "add_context_device failed.\n");
         return 1;
     }
 
diff --git a/tools/optimize/yolov5s-opt.py b/tools/optimize/yolov5s-opt.py
index 3e48ddadfc0940848bcf1efd2de63db8b42f769a..74f40cec8a9a81eb5e815a7d89fbae21230693b2 100644
--- a/tools/optimize/yolov5s-opt.py
+++ b/tools/optimize/yolov5s-opt.py
@@ -60,7 +60,7 @@ def parse_args():
 args = parse_args()
 
 
-def cut_focus_output(input_node, in_name, out_name):
+def cut_focus_output(input_node, in_name, out_name, cut_focus):
     """
     cut the focus and postprocess nodes
     Args:
@@ -98,8 +98,11 @@ def cut_focus_output(input_node, in_name, out_name):
             del input_node[i]
 
     # cut input node
-    for n in in_name:
-        new_nodes = input_node[(node_dict[n] + 1):]
+    if cut_focus:
+        for n in in_name:
+            new_nodes = input_node[(node_dict[n] + 1):]
+    else:
+        new_nodes = input_node[:]
 
     return new_nodes
 
@@ -181,9 +184,8 @@ def main():
     new_nodes = old_node[:]
 
     # cut the focus and postprocess nodes
-    if args.cut_focus:
-        print("[Quant Tools Info]: Step 1, Remove the focus and postprocess nodes.")
-        new_nodes = cut_focus_output(old_node, in_tensor, out_tensor)
+    print("[Quant Tools Info]: Step 1, Remove the focus and postprocess nodes.")
+    new_nodes = cut_focus_output(old_node, in_tensor, out_tensor, args.cut_focus)
 
     # op fusion, using HardSwish replace the Sigmoid and Mul
     print("[Quant Tools Info]: Step 2, Using hardswish replace the sigmoid and mul.")