fix bug
Signed-off-by: buzzxu <xiang.leau@gmail.com>