2 files changed, 4 insertions, 7 deletions
diff --git a/web/src/js/__tests__/components/FlowTable/FlowTableHeadSpec.js b/web/src/js/__tests__/components/FlowTable/FlowTableHeadSpec.js
index 51ff4ceb..3b7302f2 100644
--- a/web/src/js/__tests__/components/FlowTable/FlowTableHeadSpec.js
+++ b/web/src/js/__tests__/components/FlowTable/FlowTableHeadSpec.js
@@ -2,9 +2,8 @@ import React from 'react'
 import renderer from 'react-test-renderer'
 import ConnectedHead, { FlowTableHead } from '../../../components/FlowTable/FlowTableHead'
 import { Provider } from 'react-redux'
-import configureStore from 'redux-mock-store'
+import { TStore } from '../../ducks/tutils'
 
-const mockStore = configureStore()
 
 describe('FlowTableHead Component', () => {
     let sortFn = jest.fn(),
@@ -21,7 +20,7 @@ describe('FlowTableHead Component', () => {
     })
 
     it('should connect to state', () => {
-        let store = mockStore({ flows: {sort: {desc: true, column: 'PathColumn'}} }),
+        let store = TStore(),
             provider = renderer.create(
                 <Provider store={store}>
                     <ConnectedHead/>
diff --git a/web/src/js/__tests__/components/Header/ConnectionIndicatorSpec.js b/web/src/js/__tests__/components/Header/ConnectionIndicatorSpec.js
index 88527d91..45fcb5e4 100644
--- a/web/src/js/__tests__/components/Header/ConnectionIndicatorSpec.js
+++ b/web/src/js/__tests__/components/Header/ConnectionIndicatorSpec.js
@@ -3,9 +3,7 @@ import renderer from 'react-test-renderer'
 import ConnectedIndicator, { ConnectionIndicator } from '../../../components/Header/ConnectionIndicator'
 import { ConnectionState } from '../../../ducks/connection'
 import { Provider } from 'react-redux'
-import configureStore from 'redux-mock-store'
-
-const mockStore = configureStore()
+import { TStore } from '../../ducks/tutils'
 
 describe('ConnectionIndicator Component', () => {
 
@@ -45,7 +43,7 @@ describe('ConnectionIndicator Component', () => {
     })
 
     it('should connect to state', () => {
-        let store = mockStore({ connection: {state: ConnectionState.INIT} }),
+        let store = TStore(),
             provider = renderer.create(
             <Provider store={store}>
                 <ConnectedIndicator/>