Notebook with `plot3d()` have rendering issues in the docs

[tomflexcompute]

This notebook has some formatting issue in the docs, specifically in cell[6]. Not sure if it's related to plot_3d() since other notebooks with plot_3d() are rendered correctly.

@daquintero do you have any insights?