Add MHA check wrapper