diff --git a/superset-frontend/package-lock.json b/superset-frontend/package-lock.json
index 492016be4713..1613d6d89e34 100644
--- a/superset-frontend/package-lock.json
+++ b/superset-frontend/package-lock.json
@@ -64822,7 +64822,7 @@
         "core-js": "3.40.0",
         "gh-pages": "^6.3.0",
         "jquery": "^4.0.0",
-        "memoize-one": "^5.2.1",
+        "memoize-one": "^6.0.0",
         "react": "^17.0.2",
         "react-dom": "^17.0.2",
         "react-loadable": "^5.5.0",
@@ -65296,6 +65296,12 @@
       "dev": true,
       "license": "MIT"
     },
+    "packages/superset-ui-demo/node_modules/memoize-one": {
+      "version": "6.0.0",
+      "resolved": "https://registry.npmjs.org/memoize-one/-/memoize-one-6.0.0.tgz",
+      "integrity": "sha512-rkpe71W0N0c0Xz6QD0eJETuWAJGnJ9afsl1srmwPrI+yBCkge5EycXXbYRyvL29zZVUWQCY7InPRCv3GDXuZNw==",
+      "license": "MIT"
+    },
     "packages/superset-ui-demo/node_modules/schema-utils": {
       "version": "3.3.0",
       "resolved": "https://registry.npmjs.org/schema-utils/-/schema-utils-3.3.0.tgz",
diff --git a/superset-frontend/packages/superset-ui-demo/package.json b/superset-frontend/packages/superset-ui-demo/package.json
index 344e7a268f82..6dcf9689b7ee 100644
--- a/superset-frontend/packages/superset-ui-demo/package.json
+++ b/superset-frontend/packages/superset-ui-demo/package.json
@@ -45,7 +45,7 @@
     "core-js": "3.40.0",
     "gh-pages": "^6.3.0",
     "jquery": "^4.0.0",
-    "memoize-one": "^5.2.1",
+    "memoize-one": "^6.0.0",
     "react": "^17.0.2",
     "react-dom": "^17.0.2",
     "react-loadable": "^5.5.0",
diff --git a/superset-frontend/packages/superset-ui-demo/storybook/stories/plugins/plugin-chart-table/Table.stories.tsx b/superset-frontend/packages/superset-ui-demo/storybook/stories/plugins/plugin-chart-table/Table.stories.tsx
index 16d56ca19be8..f19b9122c37b 100644
--- a/superset-frontend/packages/superset-ui-demo/storybook/stories/plugins/plugin-chart-table/Table.stories.tsx
+++ b/superset-frontend/packages/superset-ui-demo/storybook/stories/plugins/plugin-chart-table/Table.stories.tsx
@@ -18,7 +18,7 @@
  */
 
 import memoizeOne from 'memoize-one';
-import { SuperChart, VizType } from '@superset-ui/core';
+import { DataRecord, SuperChart, VizType } from '@superset-ui/core';
 import TableChartPlugin, {
   TableChartProps,
 } from '@superset-ui/plugin-chart-table';
@@ -77,7 +77,7 @@ export default {
 
 new TableChartPlugin().configure({ key: VizType.Table }).register();
 
-function expandArray<T>(input: T[], targetSize: number) {
+function expandArray<T>(input: T[], targetSize: number): T[] {
   if (!input || input.length === 0) {
     throw new Error('Cannot expand an empty array');
   }
@@ -90,8 +90,14 @@ function expandArray<T>(input: T[], targetSize: number) {
 
 // memoize expanded array so to make sure we always return the same
 // data when changing page sizes
-const expandRecords = memoizeOne(expandArray);
-const expandColumns = memoizeOne(expandArray);
+const expandRecords = memoizeOne(
+  (input: DataRecord[], targetSize: number): DataRecord[] =>
+    expandArray(input, targetSize),
+);
+const expandColumns = memoizeOne(
+  (input: string[], targetSize: number): string[] =>
+    expandArray(input, targetSize),
+);
 
 /**
  * Load sample data for testing