CoefficientSplitter: fix again ReferenceGrad(Zero) (FEniCS#460)

Author: pbrubeck

ufl/algorithms/apply_coefficient_split.py

@@ -210,14 +210,14 @@ def _handle_terminal(
         c = o
         if reference_value:
             c = ReferenceValue(c)
-        for _ in range(reference_grad):
+        for k in range(reference_grad):
             c = ReferenceGrad(c)
             if isinstance(c, Zero):
                 gdim = c.ufl_shape[-1]
                 c = Zero(
-                    o.ufl_shape + (gdim,) * reference_grad,
-                    o.ufl_free_indices,
-                    o.ufl_index_dimensions,
+                    c.ufl_shape + (gdim,) * (reference_grad - k - 1),
+                    c.ufl_free_indices,
+                    c.ufl_index_dimensions,
                 )
                 break
         if restricted == "+":