Fix build errors

Signed-off-by: Lens-F <tnoah12@gmail.com>