set i_main_module properly when a submodule is validated standalone

mbj4668 · Nov 15, 2023 · 495dda0 · 495dda0
1 parent 987f59c
commit 495dda0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyang/statements.py b/pyang/statements.py
@@ -445,7 +445,8 @@ def v_init_module(ctx, stmt):
             prefix = belongs_to.search_one('prefix')
             stmt.i_modulename = belongs_to.arg
             mod = ctx.get_module(stmt.i_modulename)
-            if mod is None:
+            if mod is None or not mod.i_is_validated:
+                # this happens if a submodule is validated standalone
                 mod = stmt
         else:
             stmt.i_modulename = ""