diff --git a/parser/func_to_graph/func2graph.py b/parser/func_to_graph/func2graph.py
index ef52916..633440f 100644
--- a/parser/func_to_graph/func2graph.py
+++ b/parser/func_to_graph/func2graph.py
@@ -1,4 +1,4 @@
-#!/usr/bin/python3
+#!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 #-------------------------------------------------------------------
 # Purpose:
diff --git a/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py b/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
index e9aaef7..ae2caee 100644
--- a/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
+++ b/tests/ut/parser/testcase/onnx_parser_testcase/onnx_model/if.py
@@ -1,3 +1,10 @@
+#!/usr/bin/env python3
+# -*- coding: UTF-8 -*-
+#-------------------------------------------------------------------
+# Purpose:
+# Copyright 2021 Huawei Technologies Co., Ltd. All rights reserved.
+#-------------------------------------------------------------------
+
 # Given a bool scalar input cond.
 # return constant tensor x if cond is True, otherwise return constant tensor y.
 import numpy as np