Update foliage layer ID assignment to use tool mask check

src/viewport-three/viewport-host.ts

@@ -9699,10 +9699,9 @@ export class ViewportHost {
       tool: toolState.tool,
       referenceHeight,
       layerIndex: toolState.tool === "paint" ? toolState.layerIndex : null,
-      foliageLayerId:
+      foliageLayerId: isTerrainFoliageLayerMaskTool(toolState.tool)
-        toolState.tool === "foliagePaint" || toolState.tool === "foliageErase"
+        ? toolState.foliageLayerId
-          ? toolState.foliageLayerId
+        : null
-          : null
     });
   }