Add a missing line

Signed-off-by: Takashiro <takashiro@qq.com>
fix bugs

Signed-off-by: Takashiro <takashiro@qq.com>