Add union wrapping #593