Fix compile

(based on commit 98bad46)

Signed-off-by: Xfurry <xfurry@scriptdev2.com>