Add guard macro to avoid build break