Fix union as far as current test go.