Fix union compilation