bug fix

Signed-off-by: Peihong Guo <peihongguo@gmail.com>