From 6f6551e34157d4e2cc6f25e359ba354a92715bd4 Mon Sep 17 00:00:00 2001
From: Matthew Scroggs <matthew.w.scroggs@gmail.com>
Date: Thu, 5 Oct 2023 15:34:16 +0100
Subject: [PATCH 1/2] use index map when creating colouring

---
 bem/src/function_space.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/bem/src/function_space.rs b/bem/src/function_space.rs
index f6568fce..dae3dc9a 100644
--- a/bem/src/function_space.rs
+++ b/bem/src/function_space.rs
@@ -36,7 +36,8 @@ impl<'a, E: FiniteElement> SerialFunctionSpace<'a, E> {
             .topology()
             .entity_count(self.grid.topology().dim())
         {
-            let vs = cell_entities.row(i).unwrap();
+            let i_t = self.index_map()[i];
+            let vs = cell_entities.row(i_t).unwrap();
             let mut c = 0;
             while c < colouring.len() {
                 let mut found = false;

From 62452f0bd9d25b9cb8d1e215f52e6fe0616519fb Mon Sep 17 00:00:00 2001
From: Matthew Scroggs <matthew.w.scroggs@gmail.com>
Date: Thu, 5 Oct 2023 15:45:25 +0100
Subject: [PATCH 2/2] .grid.topology()

---
 bem/src/function_space.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/bem/src/function_space.rs b/bem/src/function_space.rs
index dae3dc9a..2136949a 100644
--- a/bem/src/function_space.rs
+++ b/bem/src/function_space.rs
@@ -36,7 +36,7 @@ impl<'a, E: FiniteElement> SerialFunctionSpace<'a, E> {
             .topology()
             .entity_count(self.grid.topology().dim())
         {
-            let i_t = self.index_map()[i];
+            let i_t = self.grid.topology().index_map()[i];
             let vs = cell_entities.row(i_t).unwrap();
             let mut c = 0;
             while c < colouring.len() {