tested and working!

Author: RafaelGB

src/components/HeaderMenu.tsx

@@ -111,6 +111,11 @@ const HeaderMenu = (headerMenuProps: HeaderMenuProps) => {
       {
         onClick: (e: any) => {
           const sortArray = generateSortedColumns(initialState, column, true);
+          // Update state
+          dispatch({
+            type: ActionTypes.SET_SORT_BY,
+            sortArray: sortArray,
+          });
           setSortBy(sortArray);
           setExpanded(false);
         },

src/components/reducers/DatabaseDispatch.tsx

@@ -435,6 +435,8 @@ export function databaseReducer(state: TableDataType, action: ActionType) {
         if (sortedColumn) {
           column.isSorted = true;
           column.isSortedDesc = sortedColumn.desc;
+        } else {
+          column.isSorted = false;
         }
         return column;
       });