1 files changed, 5 insertions, 1 deletions
diff --git a/src/sail.ml b/src/sail.ml
index e5a5071f..7b1ee56d 100644
--- a/src/sail.ml
+++ b/src/sail.ml
@@ -167,7 +167,11 @@ let main() =
       | locs -> monomorphise_ast locs ast
     in
 
-    let ast = rewrite_ast ast in
+    let ast =
+      if !Initial_check.opt_undefined_gen then
+        rewrite_undefined (rewrite_ast ast)
+      else rewrite_ast ast in
+
     let out_name = match !opt_file_out with
       | None -> fst (List.hd parsed)
       | Some f -> f ^ ".sail" in