diff --git a/.github/workflows/run_tests.yml b/.github/workflows/run_tests.yml
index e5705c8..8aacfd5 100644
--- a/.github/workflows/run_tests.yml
+++ b/.github/workflows/run_tests.yml
@@ -13,7 +13,7 @@ jobs:
       - uses: actions/checkout@v3
 
       - name: Install dependencies
-        run: python3 -m pip install pyvista
+        run: python3 -m pip install --break-system-packages pyvista
 
       - name: Run unit tests
         run: python3 -m pytest test/
diff --git a/febug/plot.py b/febug/plot.py
index ad91112..e24a826 100644
--- a/febug/plot.py
+++ b/febug/plot.py
@@ -15,6 +15,8 @@
 @functools.singledispatch
 def _to_pyvista_grid(mesh: dolfinx.mesh.Mesh, tdim: int,
                      entities=None):
+    mesh.topology.create_connectivity(0, tdim)
+    mesh.topology.create_connectivity(tdim, mesh.topology.dim)
     return pyvista.UnstructuredGrid(*dolfinx.plot.vtk_mesh(
         mesh, tdim, entities))
 
@@ -110,7 +112,7 @@ def create_plottable_ufl_expression(
 
 
 def plot_function(u: dolfinx.fem.function.Function,
-                  plotter: pyvista.Plotter=None):
+                  plotter: pyvista.Plotter=None, **pv_args):
     if plotter is None:
         plotter = pyvista.Plotter()
 
@@ -122,7 +124,8 @@ def plot_function(u: dolfinx.fem.function.Function,
         # No data on process
         return plotter
 
-    plotter.add_mesh(grid, scalars=u.name, show_scalar_bar=True)
+    pv_args.setdefault("show_scalar_bar", True)
+    plotter.add_mesh(grid, scalars=u.name, **pv_args)
 
     if mesh.geometry.dim == 2:
         plotter.enable_parallel_projection()