diff --git a/.github/workflows/test_package_openmpi.yml b/.github/workflows/test_package_openmpi.yml
index 057e607..ea167e5 100644
--- a/.github/workflows/test_package_openmpi.yml
+++ b/.github/workflows/test_package_openmpi.yml
@@ -46,7 +46,7 @@ jobs:
         run: python3 -m pip install --break-system-packages --upgrade pip setuptools
 
       - name: Install DOLFINx
-        uses: jorgensd/actions/install-dolfinx@v0.3
+        uses: jorgensd/actions/install-dolfinx@v0.4
         with:
           adios2: ${{ matrix.adios2 }}
           petsc_arch: ${{ env.PETSC_ARCH }}
diff --git a/.github/workflows/test_redhat.yml b/.github/workflows/test_redhat.yml
index ef7691f..196dfda 100644
--- a/.github/workflows/test_redhat.yml
+++ b/.github/workflows/test_redhat.yml
@@ -58,7 +58,7 @@ jobs:
         run: python3 -m pip install ${{ steps.python-version.outputs.PYTHON_FLAGS}} --upgrade pip setuptools
 
       - name: Install DOLFINx
-        uses: jorgensd/actions/install-dolfinx@v0.3
+        uses: jorgensd/actions/install-dolfinx@v0.4
         with:
           adios2: ${{ matrix.adios2 }}
           petsc_arch: ${{ env.PETSC_ARCH }}
diff --git a/pyproject.toml b/pyproject.toml
index 99234c4..f8856ef 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -3,12 +3,12 @@ requires = ["setuptools>=61.0.0", "wheel"]
 
 [project]
 name = "adios4dolfinx"
-version = "0.9.0"
+version = "0.9.0.dev0"
 description = "Checkpointing functionality for DOLFINx meshes/functions with ADIOS2"
 authors = [{ name = "Jørgen S. Dokken", email = "dokken@simula.no" }]
 license = { file = "LICENSE" }
 readme = "README.md"
-dependencies = ["fenics-dolfinx>=0.9.0"]
+dependencies = ["fenics-dolfinx>0.9.0"]
 
 [project.optional-dependencies]
 test = ["pytest", "coverage", "ipyparallel"]
diff --git a/src/adios4dolfinx/checkpointing.py b/src/adios4dolfinx/checkpointing.py
index e828e3d..ef5b653 100644
--- a/src/adios4dolfinx/checkpointing.py
+++ b/src/adios4dolfinx/checkpointing.py
@@ -653,7 +653,8 @@ def write_mesh(
         partition_processes = mesh.comm.size
 
         # Get partitioning
-        cell_map = mesh.topology.index_map(mesh.topology.dim).index_to_dest_ranks()
+        consensus_tag = 1202
+        cell_map = mesh.topology.index_map(mesh.topology.dim).index_to_dest_ranks(consensus_tag)
         num_cells_local = mesh.topology.index_map(mesh.topology.dim).size_local
         cell_offsets = cell_map.offsets[: num_cells_local + 1]
         if cell_offsets[-1] == 0: