fix compile

Signed-off-by: Whitehawkx <xxwhitehawk@gmail.com>