Type error in createServerValidate when using discriminated union schemas

import { createServerValidate, formOptions } from '@tanstack/react-form/remix';
import { z } from 'zod';

const UnionFoo = z.object({ discriminator: z.literal('foo'), foo: z.string() });

const UnionBar = z.object({ discriminator: z.literal('bar') });

const ExampleSchema = z.discriminatedUnion('discriminator', [
  UnionFoo,
  UnionBar,
]);

const formDefaultValues: z.input<typeof ExampleSchema> = {
  discriminator: 'foo',
  foo: 'example',
};

const formOpts = formOptions({
  /* Works with the casting below. Is this the preferred way? */
  defaultValues: formDefaultValues /* as z.input<typeof ExampleSchema> */,
  validators: {
    onChange: ExampleSchema,
  },
});

// Type error
const serverValidate = createServerValidate({
  ...formOpts,
  onServerValidate: (values) => {
    console.log(values);
  },
});