diff --git a/web/src/components/ui/table/AppDataGrid/TanStackDataGrid/DataGridHeader.tsx b/web/src/components/ui/table/AppDataGrid/TanStackDataGrid/DataGridHeader.tsx
index 8fc214aca..e26f3bb3b 100644
--- a/web/src/components/ui/table/AppDataGrid/TanStackDataGrid/DataGridHeader.tsx
+++ b/web/src/components/ui/table/AppDataGrid/TanStackDataGrid/DataGridHeader.tsx
@@ -7,6 +7,7 @@ import { CaretDown, CaretUp } from '../../../icons/NucleoIconFilled';
 import { HEADER_HEIGHT } from './constants';
 import { useSortColumnContext } from './SortColumnWrapper';
 import { Virtualizer } from '@tanstack/react-virtual';
+import { Text } from '@/components/ui/typography';
 
 interface DraggableHeaderProps {
   header: Header<Record<string, string | number | Date | null>, unknown>;
@@ -65,15 +66,15 @@ const DraggableHeader: React.FC<DraggableHeaderProps> = ({
       onClick={header.column.getCanSort() ? header.column.getToggleSortingHandler() : undefined}>
       <span
         className={cn(
-          'flex h-full flex-1 items-center space-x-1.5 p-2',
+          'flex h-full flex-1 items-center space-x-1.5 overflow-hidden p-2',
           draggable && 'cursor-grab'
         )}
         ref={draggable ? setDragNodeRef : undefined}
         {...attributes}
         {...listeners}>
-        <span className="text-gray-dark text-base font-normal">
+        <Text variant={'secondary'} truncate>
           {flexRender(header.column.columnDef.header, header.getContext())}
-        </span>
+        </Text>
 
         {sortable && (
           <>
diff --git a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeaderVersionHistory/VersionHistoryHeaderButtons.tsx b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeaderVersionHistory/VersionHistoryHeaderButtons.tsx
index 5748fe27b..e5f8b9453 100644
--- a/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeaderVersionHistory/VersionHistoryHeaderButtons.tsx
+++ b/web/src/layouts/ChatLayout/FileContainer/FileContainerHeader/FileContainerHeaderVersionHistory/VersionHistoryHeaderButtons.tsx
@@ -46,7 +46,7 @@ export const VersionHistoryHeaderButtons: React.FC<{}> = React.memo(({}) => {
         disabled={isSelectedVersionCurrent || !currentVersion}
         onClick={onClickRestoreVersionPreflight}
         loading={isRestoringVersion}>
-        Restore version {restoringVersion}
+        Restore version
       </Button>
     </div>
   );
diff --git a/web/src/styles/styles.scss b/web/src/styles/styles.scss
index ddb820853..c85d75560 100644
--- a/web/src/styles/styles.scss
+++ b/web/src/styles/styles.scss
@@ -45,3 +45,7 @@ body {
 p {
   @apply leading-1.5;
 }
+
+th {
+  @apply text-base font-normal;
+}