Fix compile

(based on commit 98bad46)

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

(based on commit 10823ef)

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