structure FEATURE support for augment-structure
diff --git a/src/plugins_exts/yangdata.c b/src/plugins_exts/yangdata.c
index 7a2585a..53b918d 100644
--- a/src/plugins_exts/yangdata.c
+++ b/src/plugins_exts/yangdata.c
@@ -53,7 +53,7 @@
     }
 
     /* parse yang-data substatements */
-    LY_ARRAY_CREATE_GOTO(lyplg_extp_cur_pmod(pctx)->mod->ctx, ext->substmts, 3, ret, emem);
+    LY_ARRAY_CREATE_GOTO(lyplg_ext_parse_get_cur_pmod(pctx)->mod->ctx, ext->substmts, 3, ret, emem);
     LY_ARRAY_INCREMENT(ext->substmts);
     ext->substmts[0].stmt = LY_STMT_CONTAINER;
     ext->substmts[0].storage = &ext->parsed;