diff --git a/apis/nucleus/src/hooks/useStyling.js b/apis/nucleus/src/hooks/useStyling.js
index c77f5576c..5470cd0e8 100644
--- a/apis/nucleus/src/hooks/useStyling.js
+++ b/apis/nucleus/src/hooks/useStyling.js
@@ -28,6 +28,7 @@ const THEME_OBJECT_TYPE_MAP = {
   mapchart: 'mapChart',
   bulletchart: 'bulletChart',
   'sn-table': 'straightTableV2',
+  waterfallchart: 'waterfallChart',
 };
 
 const getThemeObjectType = (visualization) => {