diff --git a/latticeCompose.py b/latticeCompose.py
index a40f13a..f2ea6a5 100644
--- a/latticeCompose.py
+++ b/latticeCompose.py
@@ -81,7 +81,7 @@ class Compose(latticeBaseFeature.LatticeFeature):
         tool = obj.Tool.Shape
         if tool.ShapeType != 'Compound':
             tool = Part.makeCompound([tool])
-        if obj.FlattenBaseHierarchy:
+        if obj.FlattenToolHierarchy:
             toolChildren = LCE.AllLeaves(tool)
         else:
             toolChildren = tool.childShapes()