Fix JSX code generator