fix bug

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