Fix bug

Signed-off-by: leo-ma <begeekmyfriend@gmail.com>