Compilation issue fix
Credit miaosp