diff --git a/src/tree_schema_compile.c b/src/tree_schema_compile.c
index 3522e9a..d29a7b4 100644
--- a/src/tree_schema_compile.c
+++ b/src/tree_schema_compile.c
@@ -642,7 +642,7 @@
 
     must->cond = lyxp_expr_parse(ctx->ctx, must_p->arg);
     LY_CHECK_ERR_GOTO(!must->cond, ret = ly_errcode(ctx->ctx), done);
-
+    must->module = ctx->mod_def;
     DUP_STRING(ctx->ctx, must_p->eapptag, must->eapptag);
     DUP_STRING(ctx->ctx, must_p->emsg, must->emsg);
     DUP_STRING(ctx->ctx, must_p->dsc, must->dsc);
diff --git a/tests/src/test_tree_schema_compile.c b/tests/src/test_tree_schema_compile.c
index 5147e01..d731c61 100644
--- a/tests/src/test_tree_schema_compile.c
+++ b/tests/src/test_tree_schema_compile.c
@@ -2944,7 +2944,7 @@
                                   "deviation /x:b {deviate add {default x:b;}}"
                                   "deviation /x:c {deviate add {default bye;}}"
                                   "deviation /x:d {deviate add {default all; default people;}}"
-                                  "deviation /x:c2 {deviate add {default hi;}}"
+                                  "deviation /x:c2 {deviate add {default hi; must 1;}}"
                                   "deviation /x:d2 {deviate add {default hi; default all;}}}", LYS_IN_YANG));
     assert_non_null((mod = ly_ctx_get_module_implemented(ctx, "e")));
     assert_non_null(node = mod->compiled->data);
@@ -2970,6 +2970,8 @@
     assert_int_equal(0, dynamic);
     assert_int_equal(6, leaf->dflt->realtype->refcount); /* 3x type reference, 3x default value reference
     - previous type's default values were replaced by node's default values where d2 now has 2 default values */
+    assert_int_equal(1, LY_ARRAY_SIZE(leaf->musts));
+    assert_ptr_equal(leaf->musts[0].module, ly_ctx_get_module_implemented(ctx, "g"));
     assert_non_null(llist = (struct lysc_node_leaflist*)leaf->next);
     assert_int_equal(2, LY_ARRAY_SIZE(llist->dflts));
     assert_string_equal("hi", llist->dflts[0]->realtype->plugin->print(llist->dflts[0], LYD_XML, NULL, NULL, &dynamic));