diff --git a/buildpacks/java/java/function.go b/buildpacks/java/java/function.go
index 43a81003..e9a1d002 100644
--- a/buildpacks/java/java/function.go
+++ b/buildpacks/java/java/function.go
@@ -95,8 +95,8 @@ func WithDefaultFunction(defaultFunctionName string, override bool, funcYamlName
 		metadata["bp-default-function"] = funcYamlName
 
 		if override {
-			fun.functionClass = defaultFunctionName
-			metadata["bp-function-class"] = defaultFunctionName
+			fun.defaultFunctionName = defaultFunctionName
+			metadata["bp-default-function"] = defaultFunctionName
 		}
 
 		fun.overrideDefaultFunctionName = override