Add conditional compile directive to avoid build error