From 1c3233cc0f0ff4daf57715c51d831409d64f9e91 Mon Sep 17 00:00:00 2001
From: Simon Michael <simon@joyful.com>
Date: Mon, 15 Oct 2018 18:30:57 -0700
Subject: [PATCH] ui: replace odd-looking pattern

---
 hledger-ui/Hledger/UI/UIState.hs | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/hledger-ui/Hledger/UI/UIState.hs b/hledger-ui/Hledger/UI/UIState.hs
index 615ddc6c5..b1470518c 100644
--- a/hledger-ui/Hledger/UI/UIState.hs
+++ b/hledger-ui/Hledger/UI/UIState.hs
@@ -113,8 +113,9 @@ toggleTree :: UIState -> UIState
 toggleTree ui@UIState{aopts=uopts@UIOpts{cliopts_=copts@CliOpts{reportopts_=ropts}}} =
   ui{aopts=uopts{cliopts_=copts{reportopts_=toggleTreeMode ropts}}}
   where
-    toggleTreeMode ropts@ReportOpts{accountlistmode_=ALTree} = ropts{accountlistmode_=ALFlat}
-    toggleTreeMode ropts = ropts{accountlistmode_=ALTree}
+    toggleTreeMode ropts
+      | accountlistmode_ ropts == ALTree = ropts{accountlistmode_=ALFlat}
+      | otherwise                        = ropts{accountlistmode_=ALTree}
 
 -- | Toggle between historical balances and period balances.
 toggleHistorical :: UIState -> UIState