diff --git a/.github/workflows/test_package_openmpi.yml b/.github/workflows/test_package_openmpi.yml
index f63147d..e851010 100644
--- a/.github/workflows/test_package_openmpi.yml
+++ b/.github/workflows/test_package_openmpi.yml
@@ -38,7 +38,7 @@ jobs:
 
     strategy:
       matrix:
-        adios2: ["default", "v2.9.2", "v2.10.0"]
+        adios2: ["default", "v2.9.2", "v2.10.1"]
     steps:
       - uses: actions/checkout@v4
 
diff --git a/.github/workflows/test_redhat.yml b/.github/workflows/test_redhat.yml
index ac21fde..d2f3c94 100644
--- a/.github/workflows/test_redhat.yml
+++ b/.github/workflows/test_redhat.yml
@@ -37,7 +37,7 @@ jobs:
 
     strategy:
       matrix:
-        adios2: ["v2.10.0"]
+        adios2: ["v2.10.1"]
     steps:
       - uses: actions/checkout@v4
 
diff --git a/src/adios4dolfinx/checkpointing.py b/src/adios4dolfinx/checkpointing.py
index 836563d..b45b73f 100644
--- a/src/adios4dolfinx/checkpointing.py
+++ b/src/adios4dolfinx/checkpointing.py
@@ -564,7 +564,7 @@ def read_mesh_data(
         )
 
         def partitioner(comm: MPI.Intracomm, n, m, topo):
-            assert len(partition_graph.offsets) - 1 == topo.num_nodes
+            assert len(topo[0]) % (len(partition_graph.offsets) - 1) == 0
             return partition_graph
     else:
         partitioner = dolfinx.cpp.mesh.create_cell_partitioner(ghost_mode)