diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 612ee5f3..db5d66ee 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -9764,6 +9764,11 @@ export class ViewportHost {
       dirtyBounds: initialStamp.dirtyBounds,
       heightSampleIndices: new Set(initialStamp.heightSampleIndices),
       paintWeightIndices: new Set(initialStamp.paintWeightIndices),
+      foliageMaskValueKeys: new Set(
+        initialStamp.foliageMaskValueIndices.map((entry) =>
+          createTerrainFoliageMaskValueKey(entry.layerId, entry.index)
+        )
+      ),
       referenceHeight,
       lastAppliedPoint: {
         x: hit.point.x,