diff --git a/compose/service/record_test.go b/compose/service/record_test.go
index 4006a3b40..7b683c229 100644
--- a/compose/service/record_test.go
+++ b/compose/service/record_test.go
@@ -819,6 +819,8 @@ func TestRecord_contextualRolesAccessControl(t *testing.T) {
 			ModuleID:    mod.ID,
 		}
 
+		rr[i].SetModule(mod)
+
 		if i%2 == 0 {
 			// let's own half of the records
 			rr[i].OwnedBy = user.ID
diff --git a/tests/compose/record_test.go b/tests/compose/record_test.go
index fe93ea115..602baa859 100644
--- a/tests/compose/record_test.go
+++ b/tests/compose/record_test.go
@@ -161,6 +161,8 @@ func (h helper) makeRecord(module *types.Module, rvs ...*types.RecordValue) *typ
 		Values: values.Formatter().Run(module, rvs),
 	}
 
+	rec.SetModule(module)
+
 	h.noError(dalutils.ComposeRecordCreate(context.Background(), defDal, module, rec))
 
 	return rec