diff --git a/install.py b/install.py
index 7a3bd4b..61b5962 100755
--- a/install.py
+++ b/install.py
@@ -56,7 +56,7 @@ def gen_col_headers(html_file, file_dict):
 
 def gen_columns(html_file, file_dict):
     for key in file_dict:
-        if key == "tree":
+        if key.split("_")[0] == "tree":
             html_file.write("<div class=\"column\">\n")
             html_file.write("  <div class=\"tree\">\n")
             html_file.write("    <h1>.</h1>\n")