Add compatibility of partitioners with unions