All Submissions:

• I installed pre-commit prior to committing my changes (see development setup docs).
• My commit messages conform to conventional commits.
• My PR title conforms to conventional commits.
• I linked to any relevant issues.

Description of changes

We remove the old style

with solara.Element() as main:
    ...
return main

since the return would skip any components not inside main.