diff --git a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt
index 0b9f17652b1..b395a0d4604 100644
--- a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt
+++ b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/EventStreamTests.kt
@@ -193,6 +193,8 @@ extension EventStreamTestClientTypes.TestStream: ClientRuntime.MessageUnmarshall
         val contents = getFileContents(context.manifest, "/Example/EventStreamTestClient.swift")
         var expected = """
 extension EventStreamTestClient: EventStreamTestClientProtocol {
+    /// Performs the `TestStreamOp` operation on the `TestService` service.
+    ///
     /// This operation is cool.
     ///
     /// - Parameter TestStreamOpInput : [no documentation found]
@@ -239,7 +241,7 @@ extension EventStreamTestClient: EventStreamTestClientProtocol {
     }
 
 }
-        """.trimIndent()
+"""
         contents.shouldContainOnlyOnce(expected)
     }
 
diff --git a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt
index 077ec4e8daa..e2fbd35882b 100644
--- a/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt
+++ b/codegen/smithy-aws-swift-codegen/src/test/kotlin/software/amazon/smithy/aws/swift/codegen/awsquery/AWSQueryOperationStackTest.kt
@@ -19,9 +19,10 @@ class AWSQueryOperationStackTest {
         val context = setupTests("awsquery/query-empty-input-output.smithy", "aws.protocoltests.query#AwsQuery")
         val contents = getFileContents(context.manifest, "/Example/QueryProtocolClient.swift")
         contents.shouldSyntacticSanityCheck()
-        val expectedContents =
-            """
+        val expectedContents = """
 extension QueryProtocolClient: QueryProtocolClientProtocol {
+    /// Performs the `NoInputAndOutput` operation on the `AwsQuery` service.
+    ///
     /// This is a very cool operation.
     ///
     /// - Parameter NoInputAndOutputInput : [no documentation found]
@@ -57,7 +58,8 @@ extension QueryProtocolClient: QueryProtocolClientProtocol {
         return result
     }
 
-}"""
+}
+"""
         contents.shouldContainOnlyOnce(expectedContents)
     }