Help with displaying Sankey Diagram correctly

My Sankey diagram shows like this:

It seems like the plotting is limited to be rectangular in the Jupyter notebook, that is why all nodes get stacked/overlapped in that way. Is this a Jupyter notebook problem? Or something I can fix with Plotly? It qould be awesome if I can fix this with plotly. I don’t mind if the plot grows vertically so that I will need to scroll. Any help is appreciated.

Have you tried increasing the figure height? (fig.update_layout(height=2000))

Thank you. It worked