diff --git a/src/viewport-three/viewport-host.ts b/src/viewport-three/viewport-host.ts
index 3a5fff97..f5bb1ba4 100644
--- a/src/viewport-three/viewport-host.ts
+++ b/src/viewport-three/viewport-host.ts
@@ -1638,6 +1638,10 @@ export class ViewportHost {
       this.cancelActiveTerrainBrushStroke(false);
     }
 
+    if (terrainChanged || toolChanged || layerChanged || terrainBrushState === null) {
+      this.rebuildDisplayedTerrainState();
+    }
+
     if (terrainBrushState === null || this.toolMode !== "select") {
       this.setTerrainBrushHover(null);
       return;