diff --git a/src/sourcetransform.rs b/src/sourcetransform.rs
index aed7cc5..37458df 100644
--- a/src/sourcetransform.rs
+++ b/src/sourcetransform.rs
@@ -662,9 +662,9 @@ mod tests {
             Duration::from_secs(2),
             client.source_transform_fn(ReceiverStream::new(rx)),
         )
-            .await
-            .map_err(|_| "timeout while getting stream for source_transform_fn")??
-            .into_inner();
+        .await
+        .map_err(|_| "timeout while getting stream for source_transform_fn")??
+        .into_inner();
 
         let handshake_resp = stream.message().await?.unwrap();
         assert!(
@@ -767,9 +767,9 @@ mod tests {
             Duration::from_secs(2),
             client.source_transform_fn(ReceiverStream::new(rx)),
         )
-            .await
-            .map_err(|_| "timeout while getting stream for source_transform_fn")??
-            .into_inner();
+        .await
+        .map_err(|_| "timeout while getting stream for source_transform_fn")??
+        .into_inner();
 
         let handshake_resp = stream.message().await?.unwrap();
         assert!(
@@ -800,4 +800,4 @@ mod tests {
         assert!(task.is_finished(), "gRPC server is still running");
         Ok(())
     }
-}
\ No newline at end of file
+}