From c0b72e70f6f61b2c2639c9b280ad4f455c32c7c4 Mon Sep 17 00:00:00 2001
From: Joe Dean <jpd62@cam.ac.uk>
Date: Thu, 7 Dec 2023 11:39:05 +0000
Subject: [PATCH] Fix array size

---
 cpp/dolfinx/fem/DofMap.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cpp/dolfinx/fem/DofMap.cpp b/cpp/dolfinx/fem/DofMap.cpp
index 07a5a9aa60e..d78b1209b96 100644
--- a/cpp/dolfinx/fem/DofMap.cpp
+++ b/cpp/dolfinx/fem/DofMap.cpp
@@ -135,7 +135,8 @@ fem::DofMap build_collapsed_dofmap(const DofMap& dofmap_view,
   ss << "new_to_old_conn = " << new_to_old_conn << "\n";
 
   // TODO Simplify
-  std::vector<std::int32_t> non_blocked_to_blocked(indices_conn.size(), -1);
+  std::vector<std::int32_t> non_blocked_to_blocked(
+    indices_conn.back() + 1, -1);
   for (std::size_t i = 0; i < indices_conn.size(); ++i)
   {
     non_blocked_to_blocked[indices_conn[i]] = dofs_view[i];