Fix loc on open of exported module

tjammer · Aug 29, 2023 · c13a58f · c13a58f
1 parent 7cfc422
commit c13a58f
Showing 1 changed file with 9 additions and 5 deletions.
diff --git a/lib/typing/env.ml b/lib/typing/env.ml
@@ -692,15 +692,19 @@ let pop_scope env =
   | ({ kind = Smodule _; _ } as hd) :: _ -> hd
   | _ -> failwith "Internal Error: Not a module scope in [pop_scope]"
 
-let open_module env loc name =
-  let scope = env.find_module env loc name |> get_module env |> snd in
-  let cont = empty_scope (Scont (Hashtbl.create 64)) in
-  { env with values = cont :: scope :: env.values }
-
 let fix_scope_loc scope loc =
   let kind =
     match scope.kind with
     | Smodule usage -> Smodule { usage with loc }
     | (Stoplevel _ | Sfunc _ | Scont _) as kind -> kind
   in
   { scope with kind }
+
+let open_module env loc name =
+  let scope =
+    env.find_module env loc name |> get_module env |> snd |> fun scope ->
+    fix_scope_loc scope loc
+  in
+
+  let cont = empty_scope (Scont (Hashtbl.create 64)) in
+  { env with values = cont :: scope :: env.values }