fixed build error

Signed-off-by: ymsquall <ymsquare@hotmail.com>