diff --git a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/cloudfrontkeyvaluestore/CloudFrontKeyValueStoreSigV4APrioritizationCustomization.kt b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/cloudfrontkeyvaluestore/CloudFrontKeyValueStoreSigV4APrioritizationCustomization.kt
index 8283133e3c2..eb324676d6d 100644
--- a/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/cloudfrontkeyvaluestore/CloudFrontKeyValueStoreSigV4APrioritizationCustomization.kt
+++ b/codegen/aws-sdk-codegen/src/main/kotlin/aws/sdk/kotlin/codegen/customization/cloudfrontkeyvaluestore/CloudFrontKeyValueStoreSigV4APrioritizationCustomization.kt
@@ -30,7 +30,7 @@ class CloudFrontKeyValueStoreSigV4APrioritizationCustomization : KotlinIntegrati
                     val builder = (shape as ServiceShape).toBuilder()
                     val authTraitValueSet = shape.getTrait<AuthTrait>()?.valueSet ?: mutableSetOf()
 
-                    if (!authTraitValueSet.contains(SigV4ATrait.ID)) {
+                    if (authTraitValueSet.firstOrNull() != SigV4ATrait.ID) {
                         // remove existing auth trait
                         builder.removeTrait(AuthTrait.ID)