diff --git a/internal/cloudapi/v2/compose_test.go b/internal/cloudapi/v2/compose_test.go
index 8446938e1..28f91829a 100644
--- a/internal/cloudapi/v2/compose_test.go
+++ b/internal/cloudapi/v2/compose_test.go
@@ -43,6 +43,13 @@ func GetTestBlueprint() blueprint.Blueprint {
 		{Name: "tmux"},
 	}
 
+	expected.EnabledModules = []blueprint.EnabledModule{
+		{
+			Name:   "node",
+			Stream: "20",
+		},
+	}
+
 	// Containers
 	expected.Containers = []blueprint.Container{
 		blueprint.Container{
@@ -207,6 +214,12 @@ func TestGetBlueprintFromCustomizations(t *testing.T) {
 				Groups:   &[]string{"users", "wheel"},
 			}},
 		Packages: &[]string{"bash", "tmux"},
+		EnabledModules: &[]Module{
+			{
+				Name:   "node",
+				Stream: "20",
+			},
+		},
 		Containers: &[]Container{
 			Container{
 				Name:   common.ToPtr("container-name"),
@@ -374,6 +387,12 @@ func TestGetBlueprintFromCompose(t *testing.T) {
 		Name:     "empty blueprint",
 		Version:  common.ToPtr("0.0.0"),
 		Packages: &[]Package{{Name: "bash"}, {Name: "tmux"}},
+		EnabledModules: &[]Module{
+			{
+				Name:   "node",
+				Stream: "20",
+			},
+		},
 		Containers: &[]Container{
 			Container{
 				Name:   common.ToPtr("container-name"),
diff --git a/internal/cloudapi/v2/v2_test.go b/internal/cloudapi/v2/v2_test.go
index 0aa47c7e0..4fbd1bc45 100644
--- a/internal/cloudapi/v2/v2_test.go
+++ b/internal/cloudapi/v2/v2_test.go
@@ -1679,6 +1679,7 @@ func TestDepsolveBlueprint(t *testing.T) {
 				"name": "deptest1",
 				"version": "0.0.1",
 				"distro": "%[1]s",
+				"enabled_modules": [{ "name": "deps", "stream": "1" }],
 				"packages": [
 					{ "name": "dep-package", "version": "*" }
 			]},