add missing key

FInAT · Apr 10, 2017 · fecac5b · fecac5b
1 parent 013cbca
commit fecac5b
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/finat/quadrature_element.py b/finat/quadrature_element.py
@@ -30,10 +30,8 @@ def degree(self):
     @cached_property
     def _entity_dofs(self):
         # Inspired by ffc/quadratureelement.py
-        entity_dofs = {
-            {entity: [] for entity in entities}
-            for dim, entities in iteritems(self.cell.get_topology())
-        }
+        entity_dofs = {dim: {entity: [] for entity in entities}
+                       for dim, entities in iteritems(self.cell.get_topology())}
         entity_dofs[self.cell.get_dimension()] = {0: list(range(self.space_dimension()))}
         return entity_dofs