fix ufc_cell

firedrakeproject · Feb 8, 2016 · 9434be8 · 9434be8
1 parent a7cc5ec
commit 9434be8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/FIAT/reference_element.py b/FIAT/reference_element.py
@@ -716,9 +716,9 @@ def ufc_cell( cell ):
     else:
         celltype = cell.cellname()
 
-    if celltype == "TensorProductCell":
-        # cell is a UFL cell
-        return TensorProductCell(ufc_cell(cell._A), ufc_cell(cell._B))
+    if " * " in celltype:
+        # Tensor product cell
+        return TensorProductCell(*map(ufc_cell, celltype.split(" * ")))
     elif celltype == "quadrilateral":
         return FiredrakeQuadrilateral()
     elif celltype == "interval":