diff --git a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/BusinessMetricsIntegration.kt b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/BusinessMetricsIntegration.kt
index 73575ddf075..012736a141d 100644
--- a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/BusinessMetricsIntegration.kt
+++ b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/BusinessMetricsIntegration.kt
@@ -30,13 +30,13 @@ class BusinessMetricsIntegration : KotlinIntegration {
     private val endpointBusinessMetricsSectionWriter = SectionWriter { writer, _ ->
         writer.write("")
         writer.write(
-            "if (endpoint.attributes.contains(#T)) request.context.#T(#T.SERVICE_ENDPOINT_OVERRIDE)",
-            RuntimeTypes.Core.BusinessMetrics.ServiceEndpointOverride,
+            "if (endpoint.attributes.contains(#T)) request.context.#T(#T.ENDPOINT_OVERRIDE)",
+            RuntimeTypes.Core.BusinessMetrics.EndpointOverride,
             RuntimeTypes.Core.BusinessMetrics.emitBusinessMetric,
             RuntimeTypes.Core.BusinessMetrics.SmithyBusinessMetric,
         )
         writer.write(
-            "if (endpoint.attributes.contains(#T)) request.context.#T(#T.ACCOUNT_ID_BASED_ENDPOINT)",
+            "if (endpoint.attributes.contains(#T)) request.context.#T(#T.ACCOUNT_ID_ENDPOINT)",
             RuntimeTypes.Core.BusinessMetrics.AccountIdBasedEndpointAccountId,
             RuntimeTypes.Core.BusinessMetrics.emitBusinessMetric,
             RuntimeTypes.Core.BusinessMetrics.SmithyBusinessMetric,