Correctly pass in stage from operations client

amzn · Nov 1, 2022 · 937724c · 937724c
1 parent 8d2dcd6
commit 937724c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Sources/SmokeAWSModelGenerate/ModelClientDelegate+addAWSClientInitializer.swift b/Sources/SmokeAWSModelGenerate/ModelClientDelegate+addAWSClientInitializer.swift
@@ -210,6 +210,7 @@ extension ModelClientDelegate {
             self.init(credentialsProvider: credentialsProvider,
                       awsRegion: awsRegion,
                       endpointHostName: endpointHostName,
+                      stage: stage,
                       endpointPort: endpointPort,
                       requiresTLS: requiresTLS,
                       service: service,
@@ -232,6 +233,7 @@ extension ModelClientDelegate {
                 credentialsProvider: credentialsProvider,
                 awsRegion: awsRegion,
                 endpointHostName: endpointHostName,
+                stage: stage,
                 endpointPort: endpointPort,
                 requiresTLS: requiresTLS,
                 service: service,