two minor fixes

src/languages/smtlib2/poly/print.ml

@@ -11,6 +11,8 @@ module Misc = Dolmen_std.Misc
 
 exception Cannot_print of string
 
+
+
 (* TODO: structured errors *)
 let _cannot_print format =
   Format.kasprintf (fun msg -> raise (Cannot_print msg)) format
@@ -922,6 +924,7 @@ module Make
 
   let attribute env fmt t =
     match S.view t with
+    | Symbol _ when is_keyword t -> sexpr env fmt t
     | App ([k; _]) when is_keyword k -> sexpr env fmt t
     | _ -> _cannot_print "not an attribtue"
 

src/languages/smtlib2/v2.6/script/print.ml

@@ -923,6 +923,7 @@ module Make
 
   let attribute env fmt t =
     match S.view t with
+    | Symbol _ when is_keyword t -> sexpr env fmt t
     | App ([k; _]) when is_keyword k -> sexpr env fmt t
     | _ -> _cannot_print "not an attribtue"
 

src/loop/transform.ml

@@ -293,9 +293,9 @@ module Smtlib2
             attrs = [];
             implicit = false;
           } in
-          add_stmt acc exit
+          add_post_logic_stmt acc exit
         in
-        let acc = add_stmt acc stmt in
+        let acc = add_post_logic_stmt acc stmt in
         flush st acc res
       end