fix bug

Signed-off-by: Wenhui Shen <swh@admpub.com>