diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 28281b7b..6106d28e 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -9560,7 +9560,11 @@ export class ViewportHost {
       settings: toolState,
       tool: toolState.tool,
       referenceHeight,
-      layerIndex: toolState.tool === "paint" ? toolState.layerIndex : null
+      layerIndex: toolState.tool === "paint" ? toolState.layerIndex : null,
+      foliageLayerId:
+        toolState.tool === "foliagePaint" || toolState.tool === "foliageErase"
+          ? toolState.foliageLayerId
+          : null
     });
   }