Bug fix

Signed-off-by: twhello <dev@twhello.com>