bug fixed

Signed-off-by: JIANG Di <dd19941105@gmail.com>