From df1efff19ba840e7a40ce01a511b85173aef7d5c Mon Sep 17 00:00:00 2001
From: Simon Howe <footless@gmail.com>
Date: Thu, 28 Jul 2016 10:55:04 +0200
Subject: [PATCH] Force a layout after a toggle to viz-mode

---
 client/app/scripts/actions/app-actions.js | 24 ++++++++++++-----------
 1 file changed, 13 insertions(+), 11 deletions(-)

diff --git a/client/app/scripts/actions/app-actions.js b/client/app/scripts/actions/app-actions.js
index 88b3b7b8d..c83a8bf48 100644
--- a/client/app/scripts/actions/app-actions.js
+++ b/client/app/scripts/actions/app-actions.js
@@ -35,17 +35,6 @@ export function toggleHelp() {
 }
 
 
-export function toggleGridMode(enabled) {
-  return (dispatch, getState) => {
-    dispatch({
-      type: ActionTypes.SET_GRID_MODE,
-      enabled
-    });
-    updateRoute(getState);
-  };
-}
-
-
 export function sortOrderChanged(sortBy, sortedDesc) {
   return (dispatch, getState) => {
     dispatch({
@@ -243,6 +232,19 @@ export function clickForceRelayout() {
   };
 }
 
+export function toggleGridMode(enabled) {
+  return (dispatch, getState) => {
+    dispatch({
+      type: ActionTypes.SET_GRID_MODE,
+      enabled
+    });
+    updateRoute(getState);
+    if (!enabled) {
+      dispatch(clickForceRelayout());
+    }
+  };
+}
+
 export function clickNode(nodeId, label, origin) {
   return (dispatch, getState) => {
     dispatch({