How to split Wizard steps into files?

[otaviobonder]

I followed the Wizard example and it's working fine. However when I try to split the steps in files, I can't get the validationSchema prop, because my direct child of Wizard is not WizardStep any more, but the step component (which has WizardStep as it's first child).

Here is a codesandbox: https://codesandbox.io/s/bold-sunset-xtx9j

How to make FirstStep clone its children?