1 files changed, 2 insertions, 6 deletions

diff --git a/ui-tree.c b/ui-tree.c
index 8c87571..26f06e3 100644
--- a/ui-tree.c
+++ b/ui-tree.c
@@ -13,7 +13,6 @@
 
 char *curr_rev;
 char *match_path;
-int header = 0;
 static int state;
 
 static void print_text_buffer(const char *name, char *buf, unsigned long size)
@@ -189,15 +188,11 @@ static void ls_head()
         html("<th class='right'>Size</th>");
         html("<th/>");
         html("</tr>\n");
-        header = 1;
 }
 
 static void ls_tail()
 {
-        if (!header)
-                return;
         html("</table>\n");
-        header = 0;
 }
 
 static void ls_tree(const unsigned char *sha1, char *path)
@@ -286,5 +281,6 @@ void cgit_print_tree(const char *rev, char *path)
         match_path = path;
         state = 0;
         read_tree_recursive(commit->tree, "", 0, 0, &paths, walk_tree, NULL);
-        ls_tail();
+        if (state == 1)
+                ls_tail();
 }