update tests to take into account new restriction on unions in #596