diff --git a/pkg/compose/relay_test.go b/pkg/compose/relay_test.go
index 34e84e1b5..7743a64df 100644
--- a/pkg/compose/relay_test.go
+++ b/pkg/compose/relay_test.go
@@ -110,18 +110,18 @@ func TestRelayNetworks(t *testing.T) {
 		Services: types.Services{
 			"db": db,
 			"app": {
-				Name:      "app",
-				DependsOn: types.DependsOnConfig{"db": {}},
-				Networks:  map[string]*types.ServiceNetworkConfig{"backend": nil},
+				Name:          "app",
+				WorkloadSpec:  types.WorkloadSpec{DependsOn: types.DependsOnConfig{"db": {}}},
+				ContainerSpec: types.ContainerSpec{Networks: map[string]*types.ServiceNetworkConfig{"backend": nil}},
 			},
 			"web": {
-				Name:      "web",
-				DependsOn: types.DependsOnConfig{"db": {}},
-				Networks:  map[string]*types.ServiceNetworkConfig{"frontend": nil, "backend": nil},
+				Name:          "web",
+				WorkloadSpec:  types.WorkloadSpec{DependsOn: types.DependsOnConfig{"db": {}}},
+				ContainerSpec: types.ContainerSpec{Networks: map[string]*types.ServiceNetworkConfig{"frontend": nil, "backend": nil}},
 			},
 			"other": {
-				Name:     "other",
-				Networks: map[string]*types.ServiceNetworkConfig{"private": nil},
+				Name:          "other",
+				ContainerSpec: types.ContainerSpec{Networks: map[string]*types.ServiceNetworkConfig{"private": nil}},
 			},
 		},
 		Networks: types.Networks{"default": {}, "backend": {}, "frontend": {}, "private": {}},
diff --git a/pkg/compose/start_test.go b/pkg/compose/start_test.go
index b392112a0..5f9aebd22 100644
--- a/pkg/compose/start_test.go
+++ b/pkg/compose/start_test.go
@@ -255,7 +255,7 @@ func TestStartService_PreStartSkippedWhenLowestIsRelay(t *testing.T) {
 	project := &types.Project{Name: "prj"}
 	service := types.ServiceConfig{
 		Name:     "db",
-		PreStart: []types.ServiceHook{{Command: types.ShellCommand{"init"}}},
+		PreStart: []types.PreStartHook{{ContainerSpec: types.ContainerSpec{Command: types.ShellCommand{"init"}}}},
 	}
 	relay := serviceContainer("db", 1, container.StateExited)
 	relay.Labels[api.RelayLabel] = "abc123"