[Bug fix] : fix bug.

Signed-off-by: Euhow <euhowzy@gmail.com>