diff --git a/src/Aspire.Hosting.AWS/CloudFormation/CloudFormationExtensions.cs b/src/Aspire.Hosting.AWS/CloudFormation/CloudFormationExtensions.cs
index 530d2c69e9..5e89462c61 100644
--- a/src/Aspire.Hosting.AWS/CloudFormation/CloudFormationExtensions.cs
+++ b/src/Aspire.Hosting.AWS/CloudFormation/CloudFormationExtensions.cs
@@ -127,9 +127,7 @@ public static IResourceBuilder<TDestination> WithReference<TDestination>(this IR
     {
         cloudFormationResourceBuilder.WithAnnotation(new CloudFormationReferenceAnnotation(builder.Resource));
 
-        return builder;
-
-        /*builder.WithEnvironment(async context =>
+        builder.WithEnvironment(async context =>
         {
             if (context.ExecutionContext.IsPublishMode)
             {
@@ -161,6 +159,6 @@ public static IResourceBuilder<TDestination> WithReference<TDestination>(this IR
             }
         });
 
-        return builder.WithAWSSDKConfig(cloudFormationResourceBuilder);*/
+        return builder.WithAWSSDKConfig(cloudFormationResourceBuilder);
     }
 }