diff --git a/test/test_ufl_wrapper.py b/test/test_ufl_wrapper.py
index 22257001b..8c194414b 100644
--- a/test/test_ufl_wrapper.py
+++ b/test/test_ufl_wrapper.py
@@ -25,7 +25,7 @@ def test_finite_element(inputs):
 def test_vector_element(inputs):
     e = basix.ufl.element(*inputs, rank=1)
     table = e.tabulate(0, [[0, 0]])
-    assert table.shape == (1, 1, e.dim)
+    assert table.shape == (1, 1, e.value_size, e.dim)
 
 
 @pytest.mark.parametrize("inputs", [