From 246cf659fd4bf1fb5523ddaac47a7bfef406f640 Mon Sep 17 00:00:00 2001
From: Helge Gehring <42973196+HelgeGehring@users.noreply.github.com>
Date: Wed, 10 Jan 2024 09:40:32 -0800
Subject: [PATCH] fix split_bases and split_indices for ElementVector

---
 skfem/assembly/basis/abstract_basis.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/skfem/assembly/basis/abstract_basis.py b/skfem/assembly/basis/abstract_basis.py
index 8b21baa7..1a1c7962 100644
--- a/skfem/assembly/basis/abstract_basis.py
+++ b/skfem/assembly/basis/abstract_basis.py
@@ -338,7 +338,7 @@ def split_indices(self) -> List[ndarray]:
                                e.interior_dofs])
             return output
         elif isinstance(self.elem, ElementVector):
-            ndims = self.mesh.dim()
+            ndims = self.elem.dim
             e = self.elem.elem
             for k in range(ndims):
                 output.append(np.concatenate((
@@ -359,7 +359,7 @@ def split_bases(self) -> List['AbstractBasis']:
         elif isinstance(self.elem, ElementVector):
             return [type(self)(self.mesh, self.elem.elem, self.mapping,
                                quadrature=self.quadrature)
-                    for _ in range(self.mesh.dim())]
+                    for _ in range(self.elem.dim)]
         return [self]
 
     @property