fix a bug

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