diff --git a/packages/data-viz/src/core/HeatmapChart/index.stories.tsx b/packages/data-viz/src/core/HeatmapChart/index.stories.tsx
index fe8713dd3..e62f527c6 100644
--- a/packages/data-viz/src/core/HeatmapChart/index.stories.tsx
+++ b/packages/data-viz/src/core/HeatmapChart/index.stories.tsx
@@ -159,9 +159,6 @@ export default {
   },
   component: HeatmapChart,
   parameters: {
-    // This parameter is used to set the story's layout to fullscreen
-    // and remove the padding around the storybook root element
-    layout: "fullscreen",
     snapshot: {
       skip: true,
     },
@@ -227,6 +224,9 @@ export const HeatmapDemo = {
         "tooltip",
       ],
     },
+    // This parameter is used to set the story's layout to fullscreen
+    // and remove the padding around the storybook root element
+    layout: "fullscreen",
     snapshot: {
       skip: true,
     },
diff --git a/packages/data-viz/src/core/HeatmapChart/storyUtils.ts b/packages/data-viz/src/core/HeatmapChart/storyUtils.ts
index 572a53ef3..88bea9f6d 100644
--- a/packages/data-viz/src/core/HeatmapChart/storyUtils.ts
+++ b/packages/data-viz/src/core/HeatmapChart/storyUtils.ts
@@ -3,234 +3,263 @@
  * Colors are generated using the following tool:
  * https://observablehq.com/@d3/color-schemes
  *
- * Magma Color Scheme
+ * YlOrRd Color Scheme
  */
 export const COLORS = [
-  "#19103f",
-  "#1a1042",
-  "#1c1044",
-  "#1d1147",
-  "#1e1149",
-  "#20114b",
-  "#21114e",
-  "#221150",
-  "#241253",
-  "#251255",
-  "#271258",
-  "#29115a",
-  "#2a115c",
-  "#2c115f",
-  "#2d1161",
-  "#2f1163",
-  "#311165",
-  "#331067",
-  "#341069",
-  "#36106b",
-  "#38106c",
-  "#390f6e",
-  "#3b0f70",
-  "#3d0f71",
-  "#3f0f72",
-  "#400f74",
-  "#420f75",
-  "#440f76",
-  "#451077",
-  "#471078",
-  "#491078",
-  "#4a1079",
-  "#4c117a",
-  "#4e117b",
-  "#4f127b",
-  "#51127c",
-  "#52137c",
-  "#54137d",
-  "#56147d",
-  "#57157e",
-  "#59157e",
-  "#5a167e",
-  "#5c167f",
-  "#5d177f",
-  "#5f187f",
-  "#601880",
-  "#621980",
-  "#641a80",
-  "#651a80",
-  "#671b80",
-  "#681c81",
-  "#6a1c81",
-  "#6b1d81",
-  "#6d1d81",
-  "#6e1e81",
-  "#701f81",
-  "#721f81",
-  "#732081",
-  "#752181",
-  "#762181",
-  "#782281",
-  "#792282",
-  "#7b2382",
-  "#7c2382",
-  "#7e2482",
-  "#802582",
-  "#812581",
-  "#832681",
-  "#842681",
-  "#862781",
-  "#882781",
-  "#892881",
-  "#8b2981",
-  "#8c2981",
-  "#8e2a81",
-  "#902a81",
-  "#912b81",
-  "#932b80",
-  "#942c80",
-  "#962c80",
-  "#982d80",
-  "#992d80",
-  "#9b2e7f",
-  "#9c2e7f",
-  "#9e2f7f",
-  "#a02f7f",
-  "#a1307e",
-  "#a3307e",
-  "#a5317e",
-  "#a6317d",
-  "#a8327d",
-  "#aa337d",
-  "#ab337c",
-  "#ad347c",
-  "#ae347b",
-  "#b0357b",
-  "#b2357b",
-  "#b3367a",
-  "#b5367a",
-  "#b73779",
-  "#b83779",
-  "#ba3878",
-  "#bc3978",
-  "#bd3977",
-  "#bf3a77",
-  "#c03a76",
-  "#c23b75",
-  "#c43c75",
-  "#c53c74",
-  "#c73d73",
-  "#c83e73",
-  "#ca3e72",
-  "#cc3f71",
-  "#cd4071",
-  "#cf4070",
-  "#d0416f",
-  "#d2426f",
-  "#d3436e",
-  "#d5446d",
-  "#d6456c",
-  "#d8456c",
-  "#d9466b",
-  "#db476a",
-  "#dc4869",
-  "#de4968",
-  "#df4a68",
-  "#e04c67",
-  "#e24d66",
-  "#e34e65",
-  "#e44f64",
-  "#e55064",
-  "#e75263",
-  "#e85362",
-  "#e95462",
-  "#ea5661",
-  "#eb5760",
-  "#ec5860",
-  "#ed5a5f",
-  "#ee5b5e",
-  "#ef5d5e",
-  "#f05f5e",
-  "#f1605d",
-  "#f2625d",
-  "#f2645c",
-  "#f3655c",
-  "#f4675c",
-  "#f4695c",
-  "#f56b5c",
-  "#f66c5c",
-  "#f66e5c",
-  "#f7705c",
-  "#f7725c",
-  "#f8745c",
-  "#f8765c",
-  "#f9785d",
-  "#f9795d",
-  "#f97b5d",
-  "#fa7d5e",
-  "#fa7f5e",
-  "#fa815f",
-  "#fb835f",
-  "#fb8560",
-  "#fb8761",
-  "#fc8961",
-  "#fc8a62",
-  "#fc8c63",
-  "#fc8e64",
-  "#fc9065",
-  "#fd9266",
-  "#fd9467",
-  "#fd9668",
-  "#fd9869",
-  "#fd9a6a",
-  "#fd9b6b",
-  "#fe9d6c",
-  "#fe9f6d",
-  "#fea16e",
-  "#fea36f",
-  "#fea571",
-  "#fea772",
-  "#fea973",
-  "#feaa74",
-  "#feac76",
-  "#feae77",
-  "#feb078",
-  "#feb27a",
-  "#feb47b",
-  "#feb67c",
-  "#feb77e",
-  "#feb97f",
-  "#febb81",
-  "#febd82",
-  "#febf84",
-  "#fec185",
-  "#fec287",
-  "#fec488",
-  "#fec68a",
-  "#fec88c",
-  "#feca8d",
-  "#fecc8f",
-  "#fecd90",
-  "#fecf92",
-  "#fed194",
-  "#fed395",
-  "#fed597",
-  "#fed799",
-  "#fed89a",
-  "#fdda9c",
-  "#fddc9e",
-  "#fddea0",
-  "#fde0a1",
-  "#fde2a3",
-  "#fde3a5",
-  "#fde5a7",
-  "#fde7a9",
-  "#fde9aa",
-  "#fdebac",
-  "#fcecae",
-  "#fceeb0",
-  "#fcf0b2",
-  "#fcf2b4",
-  "#fcf4b6",
-  "#fcf6b8",
-  "#fcf7b9",
-  "#fcf9bb",
-  "#fcfbbd",
-  "#fcfdbf",
+  "#ffffcc",
+  "#fffecb",
+  "#fffec9",
+  "#fffdc8",
+  "#fffdc6",
+  "#fffcc5",
+  "#fffcc4",
+  "#fffbc2",
+  "#fffac1",
+  "#fffac0",
+  "#fff9be",
+  "#fff9bd",
+  "#fff8bb",
+  "#fff8ba",
+  "#fff7b9",
+  "#fff6b7",
+  "#fff6b6",
+  "#fff5b5",
+  "#fff5b3",
+  "#fff4b2",
+  "#fff4b0",
+  "#fff3af",
+  "#fff2ae",
+  "#fff2ac",
+  "#fff1ab",
+  "#fff1aa",
+  "#fff0a8",
+  "#fff0a7",
+  "#ffefa6",
+  "#ffeea4",
+  "#ffeea3",
+  "#ffeda2",
+  "#ffeda0",
+  "#ffec9f",
+  "#ffeb9d",
+  "#ffeb9c",
+  "#ffea9b",
+  "#ffea99",
+  "#ffe998",
+  "#ffe897",
+  "#ffe895",
+  "#ffe794",
+  "#ffe693",
+  "#ffe691",
+  "#ffe590",
+  "#ffe48f",
+  "#ffe48d",
+  "#ffe38c",
+  "#fee28b",
+  "#fee289",
+  "#fee188",
+  "#fee087",
+  "#fee085",
+  "#fedf84",
+  "#fede83",
+  "#fedd82",
+  "#fedc80",
+  "#fedc7f",
+  "#fedb7e",
+  "#feda7c",
+  "#fed97b",
+  "#fed87a",
+  "#fed778",
+  "#fed777",
+  "#fed676",
+  "#fed574",
+  "#fed473",
+  "#fed372",
+  "#fed270",
+  "#fed16f",
+  "#fed06e",
+  "#fecf6c",
+  "#fece6b",
+  "#fecd6a",
+  "#fecb69",
+  "#feca67",
+  "#fec966",
+  "#fec865",
+  "#fec764",
+  "#fec662",
+  "#fec561",
+  "#fec460",
+  "#fec25f",
+  "#fec15e",
+  "#fec05c",
+  "#febf5b",
+  "#febe5a",
+  "#febd59",
+  "#febb58",
+  "#feba57",
+  "#feb956",
+  "#feb855",
+  "#feb754",
+  "#feb553",
+  "#feb452",
+  "#feb351",
+  "#feb250",
+  "#feb14f",
+  "#feb04e",
+  "#feae4d",
+  "#fead4d",
+  "#feac4c",
+  "#feab4b",
+  "#feaa4a",
+  "#fea84a",
+  "#fea749",
+  "#fea648",
+  "#fea547",
+  "#fea347",
+  "#fea246",
+  "#fea145",
+  "#fda045",
+  "#fd9e44",
+  "#fd9d44",
+  "#fd9c43",
+  "#fd9b42",
+  "#fd9942",
+  "#fd9841",
+  "#fd9741",
+  "#fd9540",
+  "#fd9440",
+  "#fd923f",
+  "#fd913f",
+  "#fd8f3e",
+  "#fd8e3e",
+  "#fd8d3d",
+  "#fd8b3c",
+  "#fd893c",
+  "#fd883b",
+  "#fd863b",
+  "#fd853a",
+  "#fd833a",
+  "#fd8139",
+  "#fd8039",
+  "#fd7e38",
+  "#fd7c38",
+  "#fd7b37",
+  "#fd7937",
+  "#fd7736",
+  "#fc7535",
+  "#fc7335",
+  "#fc7234",
+  "#fc7034",
+  "#fc6e33",
+  "#fc6c33",
+  "#fc6a32",
+  "#fc6832",
+  "#fb6731",
+  "#fb6531",
+  "#fb6330",
+  "#fb6130",
+  "#fb5f2f",
+  "#fa5d2e",
+  "#fa5c2e",
+  "#fa5a2d",
+  "#fa582d",
+  "#f9562c",
+  "#f9542c",
+  "#f9522b",
+  "#f8512b",
+  "#f84f2a",
+  "#f74d2a",
+  "#f74b29",
+  "#f64929",
+  "#f64828",
+  "#f54628",
+  "#f54427",
+  "#f44227",
+  "#f44127",
+  "#f33f26",
+  "#f23d26",
+  "#f23c25",
+  "#f13a25",
+  "#f03824",
+  "#f03724",
+  "#ef3524",
+  "#ee3423",
+  "#ed3223",
+  "#ed3123",
+  "#ec2f22",
+  "#eb2e22",
+  "#ea2c22",
+  "#e92b22",
+  "#e92921",
+  "#e82821",
+  "#e72621",
+  "#e62521",
+  "#e52420",
+  "#e42220",
+  "#e32120",
+  "#e22020",
+  "#e11f20",
+  "#e01d20",
+  "#df1c20",
+  "#de1b20",
+  "#dd1a20",
+  "#dc1920",
+  "#db1820",
+  "#da1720",
+  "#d91620",
+  "#d81520",
+  "#d71420",
+  "#d51320",
+  "#d41221",
+  "#d31121",
+  "#d21021",
+  "#d10f21",
+  "#cf0e21",
+  "#ce0d21",
+  "#cd0d22",
+  "#cc0c22",
+  "#ca0b22",
+  "#c90a22",
+  "#c80a22",
+  "#c60923",
+  "#c50823",
+  "#c40823",
+  "#c20723",
+  "#c10723",
+  "#bf0624",
+  "#be0624",
+  "#bc0524",
+  "#bb0524",
+  "#b90424",
+  "#b80424",
+  "#b60425",
+  "#b50325",
+  "#b30325",
+  "#b10325",
+  "#b00225",
+  "#ae0225",
+  "#ac0225",
+  "#ab0225",
+  "#a90125",
+  "#a70126",
+  "#a50126",
+  "#a40126",
+  "#a20126",
+  "#a00126",
+  "#9e0126",
+  "#9c0026",
+  "#9a0026",
+  "#990026",
+  "#970026",
+  "#950026",
+  "#930026",
+  "#910026",
+  "#8f0026",
+  "#8d0026",
+  "#8b0026",
+  "#8a0026",
+  "#880026",
+  "#860026",
+  "#840026",
+  "#820026",
+  "#800026",
 ];