diff --git a/operator/apis/lifecycle/v1alpha3/keptntaskdefinition_webhook_test.go b/operator/apis/lifecycle/v1alpha3/keptntaskdefinition_webhook_test.go
index efd372c307..b0db12eca9 100644
--- a/operator/apis/lifecycle/v1alpha3/keptntaskdefinition_webhook_test.go
+++ b/operator/apis/lifecycle/v1alpha3/keptntaskdefinition_webhook_test.go
@@ -15,7 +15,7 @@ import (
 func TestKeptnTaskDefinition_ValidateFields(t *testing.T) {
 
 	specWithFunctionAndContainer := KeptnTaskDefinitionSpec{
-		Function:  &FunctionSpec{},
+		Function:  &RuntimeSpec{},
 		Container: &ContainerSpec{},
 	}
 
@@ -59,7 +59,7 @@ func TestKeptnTaskDefinition_ValidateFields(t *testing.T) {
 		{
 			name: "with-function-only",
 			spec: KeptnTaskDefinitionSpec{
-				Function: &FunctionSpec{},
+				Function: &RuntimeSpec{},
 			},
 			verb: "create",
 		},