修改bug

Signed-off-by: ghuazh <ghuazh@gmail.com>