Fix union schema + primitive type validation