diff --git a/codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/EndpointGenerator.java b/codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/EndpointGenerator.java
index cbb76c9df8d57..08058e4a413b2 100644
--- a/codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/EndpointGenerator.java
+++ b/codegen/smithy-aws-typescript-codegen/src/main/java/software/amazon/smithy/aws/typescript/codegen/EndpointGenerator.java
@@ -164,7 +164,7 @@ private void writeEndpointProviderFunction() {
     private void writePartitionEndpointResolver(Partition partition) {
         OptionalUtils.ifPresentOrElse(
                 partition.getPartitionEndpoint(),
-                name -> writer.write("regionInfo = defaultEndpointProvider($S);", name),
+                name -> writer.write("return defaultRegionInfoProvider($S);", name),
                 () -> {
                     writer.openBlock("regionInfo = {", "};", () -> {
                         String template = partition.templateVariableName;