Fix one more wrap

CDCgov · Sep 26, 2024 · cf0b98b · cf0b98b
1 parent 5acf6f2
commit cf0b98b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/packages/core/components/DataTable/DataTable.tsx b/packages/core/components/DataTable/DataTable.tsx
@@ -20,6 +20,7 @@ import removeNullColumns from './helpers/removeNullColumns'
 import { TableConfig } from './types/TableConfig'
 import { Column } from '../../types/Column'
 import { pivotData } from '../../helpers/pivotData'
+import { isLegendWrapViewport } from '@cdc/core/helpers/viewports'
 
 export type DataTableProps = {
  applyLegendToRow?: Function
@@ -211,7 +212,7 @@ const DataTable = (props: DataTableProps) => {
  const getMediaControlsClasses = belowTable => {
  const classes = ['download-links']
  if (!belowTable) {
- const isLegendOnBottom = config?.legend?.position === 'bottom' || ['sm', 'xs', 'xxs'].includes(viewport)
+ const isLegendOnBottom = config?.legend?.position === 'bottom' || isLegendWrapViewport(viewport)
  if (config.brush?.active && !isLegendOnBottom) classes.push('brush-active')
  if (config.brush?.active && config.legend.hide) classes.push('brush-active')
  } else {