From a768506945a56ed1503cba230fb79d7beaa4dacf Mon Sep 17 00:00:00 2001
From: julienmalard <julien.malard@mail.mcgill.ca>
Date: Mon, 17 Aug 2020 08:23:44 -0400
Subject: [PATCH] Erez's idea

---
 lark/reconstruct.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/lark/reconstruct.py b/lark/reconstruct.py
index dfdaae1..1091681 100644
--- a/lark/reconstruct.py
+++ b/lark/reconstruct.py
@@ -198,10 +198,9 @@ class Reconstructor:
                 yield item
 
     def reconstruct(self, tree, postproc=None):
-        if postproc is None:
-            x = self._reconstruct(tree)
-        else:
-            x = postproc(self._reconstruct(tree))
+        x = self._reconstruct(tree)
+        if postproc:
+            x = postproc(x)
         y = []
         prev_item = ''
         for item in x: