1 files changed, 6 insertions, 5 deletions
diff --git a/web/src/js/__tests__/ducks/ui.js b/web/src/js/__tests__/ducks/ui.js
index ee1a008d..81ae852c 100644
--- a/web/src/js/__tests__/ducks/ui.js
+++ b/web/src/js/__tests__/ducks/ui.js
@@ -1,7 +1,8 @@
 jest.unmock("../../ducks/ui");
+jest.unmock("../../ducks/flows");
 
-import reducer, {setActiveMenu, SELECT_FLOW} from '../../ducks/ui';
-
+import reducer, {setActiveMenu} from '../../ducks/ui';
+import {SELECT_FLOW} from '../../ducks/flows';
 
 describe("ui reducer", () => {
     it("should return the initial state", () => {
@@ -10,21 +11,21 @@ describe("ui reducer", () => {
     it("should return the state for view", () => {
         expect(reducer(undefined, setActiveMenu('View'))).toEqual({ activeMenu: 'View'})
     }),
-    it("should change the state to Start", () => {
+    it("should change the state to Start when deselecting a flow and we a currently at the flow tab", () => {
         expect(reducer({activeMenu: 'Flow'},
             { type: SELECT_FLOW,
               currentSelection: '1',
               flowId : undefined
             })).toEqual({ activeMenu: 'Start'})
     }),
-    it("should change the state to Flow", () => {
+    it("should change the state to Flow when we selected a flow and no flow was selected before", () => {
         expect(reducer({activeMenu: 'Start'},
             { type: SELECT_FLOW,
               currentSelection: undefined,
               flowId : '1'
             })).toEqual({ activeMenu: 'Flow'})
     }),
-    it("should not change the state", () => {
+    it("should not change the state to Flow when OPTIONS tab is selected and we selected a flow and a flow as selected before", () => {
         expect(reducer({activeMenu: 'Options'},
             { type: SELECT_FLOW,
               currentSelection: '1',