Refactor: Change AST type from a union to a struct

1 files changed, 2 insertions, 2 deletions

diff --git a/src/optimize.ml b/src/optimize.ml
index 8c456a1b..fccf0f3a 100644
--- a/src/optimize.ml
+++ b/src/optimize.ml
@@ -65,7 +65,7 @@ let split_at_function id defs =
   | Some (pre_defs, def, post_defs) ->
      Some (List.rev pre_defs, def, post_defs)
 
-let recheck (Defs defs) =
+let recheck ({ defs; _} as ast) =
   let defs = Type_check.check_with_envs Type_check.initial_env defs in
 
   let rec find_optimizations = function
@@ -112,4 +112,4 @@ let recheck (Defs defs) =
     | [] -> []
   in
 
-  Defs (find_optimizations defs)
+  { ast with defs = find_optimizations defs }