Fix bug

Signed-off-by: Thynson <lanxingcan@gmail.com>