From a06e0fcd74837cd75f7f9c99ba08e53c19be401b Mon Sep 17 00:00:00 2001
From: gargroh <42495927+gargroh@users.noreply.github.com>
Date: Thu, 7 Nov 2019 13:13:50 +0530
Subject: [PATCH] Update useColumnVisibility.js

---
 src/plugin-hooks/useColumnVisibility.js | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/src/plugin-hooks/useColumnVisibility.js b/src/plugin-hooks/useColumnVisibility.js
index 2146170..a718965 100644
--- a/src/plugin-hooks/useColumnVisibility.js
+++ b/src/plugin-hooks/useColumnVisibility.js
@@ -1,9 +1,9 @@
 import React from 'react';
 import PropTypes from 'prop-types';
 
-import { defaultState } from '../../react-table/hooks/useTable';
-import { addActions, actions } from '../../react-table/actions';
-import { determineHeaderVisibility } from '../../react-table/utils';
+import { defaultState } from '..//hooks/useTable';
+import { addActions, actions } from '../actions';
+import { determineHeaderVisibility } from '../utils';
 
 addActions('setColumnVisibility');
 
@@ -12,6 +12,9 @@ defaultState.hiddenColumns = [];
 const propTypes = {};
 
 export const useColumnVisibility = (hooks) => {
+  hooks.columnsBeforeHeaderGroupsDeps.push((deps, instance) => {
+    return [...deps, instance.state.hiddenColumns];
+  });
   hooks.useBeforeDimensions.push(useBeforeDimensions);
 };
 
@@ -31,7 +34,6 @@ function useBeforeDimensions(instance) {
       return setState((old) => {
         return {
           ...old,
-          columnOrder: [...old.columnOrder], // FixMe: hack
           hiddenColumns: typeof updater === 'function' ? updater(old.hiddenColumns) : updater
         };
       }, actions.setColumnVisibility);