diff --git a/typescript/packages/subsurface-viewer/src/layers/grid3d/privateGrid3dLayer.ts b/typescript/packages/subsurface-viewer/src/layers/grid3d/privateGrid3dLayer.ts
index 5533ddd63..de31318af 100644
--- a/typescript/packages/subsurface-viewer/src/layers/grid3d/privateGrid3dLayer.ts
+++ b/typescript/packages/subsurface-viewer/src/layers/grid3d/privateGrid3dLayer.ts
@@ -415,6 +415,7 @@ export default class PrivateLayer extends Layer<PrivateLayerProps> {
 
         const colormap = context.device.createTexture({
             ...textureProps,
+            height: 1,
             data: data as TextureData,
         });
         return colormap;
diff --git a/typescript/packages/subsurface-viewer/src/storybook/examples/__image_snapshots__/subsurfaceviewer-examples--mixed-layer-definitions.png b/typescript/packages/subsurface-viewer/src/storybook/examples/__image_snapshots__/subsurfaceviewer-examples--mixed-layer-definitions.png
index 69afab95f..bf2f44095 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/examples/__image_snapshots__/subsurfaceviewer-examples--mixed-layer-definitions.png and b/typescript/packages/subsurface-viewer/src/storybook/examples/__image_snapshots__/subsurfaceviewer-examples--mixed-layer-definitions.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/Grid3DLayer.stories.tsx b/typescript/packages/subsurface-viewer/src/storybook/layers/Grid3DLayer.stories.tsx
index c6e2810ec..db2f688b1 100644
--- a/typescript/packages/subsurface-viewer/src/storybook/layers/Grid3DLayer.stories.tsx
+++ b/typescript/packages/subsurface-viewer/src/storybook/layers/Grid3DLayer.stories.tsx
@@ -296,7 +296,7 @@ const layerArrays = {
         pointsData: new Float32Array(gridPoints),
         polysData: new Uint32Array(gridPolys),
         propertiesData: new Uint16Array(gridProps),
-        colorMapFunction: new Uint16Array(CATEGORICAL_COLOR_TABLE.flat()),
+        colorMapFunction: new Uint8Array(CATEGORICAL_COLOR_TABLE.flat()),
     },
 };
 
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--continuous-property.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--continuous-property.png
new file mode 100644
index 000000000..e1d227c22
Binary files /dev/null and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--continuous-property.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--custom-color-func-with-clamping.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--custom-color-func-with-clamping.png
index 017d977b8..493de6570 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--custom-color-func-with-clamping.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--custom-color-func-with-clamping.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--discrete-property.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--discrete-property.png
index cad22cc35..e7b4f02f3 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--discrete-property.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--discrete-property.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--polyhedral-cells.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--polyhedral-cells.png
index 5c39aeda3..0a7cad254 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--polyhedral-cells.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--polyhedral-cells.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-2-x.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-2-x.png
index cb14f8974..3c29db7a5 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-2-x.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-2-x.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-4-x.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-4-x.png
index 692fc2ae3..7b3687fb4 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-4-x.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-4-x.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-8-x-i-jonly.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-8-x-i-jonly.png
index 764918c49..7dc3c5f61 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-8-x-i-jonly.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-8-x-i-jonly.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-array-input.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-array-input.png
index 96aeae51b..e5d822d1b 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-array-input.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid-array-input.png differ
diff --git a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid.png b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid.png
index 6b6121204..d6d8c5862 100644
Binary files a/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid.png and b/typescript/packages/subsurface-viewer/src/storybook/layers/__image_snapshots__/subsurfaceviewer-grid3d-layer--simgrid.png differ