Fix a bug in the compilation of * tests.

[tatoo.git] / src / html.ml

diff --git a/src/html.ml b/src/html.ml
index f049195..236ee39 100644 (file)
--- a/src/html.ml
+++ b/src/html.ml
@@ -87,8 +87,13 @@ let gen_trace (type s) = fun auto t tree ->
       in
       let pc = if !max_round == 0 then 1. else float_of_int last_round /. float_of_int !max_round in
       let scolor, tcolor =
-        let { sat ; todo; _ } = List.hd(M.find last_round m) in
-        let c = (last_round, sat, todo) in color c, text_color c
+        let { sat ; todo; _ } =
+          match M.find last_round m with
+            [] -> { sat = StateSet.empty; todo= StateSet.empty; msg = "ERROR" }
+          | [ e ] -> e
+          | l -> List.hd (List.tl (List.rev l))
+        in
+        let c = (last_round, StateSet.union sat todo, StateSet.empty) in color c, text_color c
       in
       let tag = QName.to_string (T.tag tree node) in
       let lbox = (String.length tag + 2) * 10 in