diff --git a/dist/index.js b/dist/index.js
index 9f35de8..ed5d8bf 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -134,7 +134,9 @@ async function run() {
     core.exportVariable('AWS_REGION', region);
 
     // Get the AWS account ID
-    const sts = new aws.STS();
+    const sts = new aws.STS({
+      customUserAgent: 'configure-aws-credentials-for-github-actions'
+    });
     const identity = await sts.getCallerIdentity().promise();
     const accountId = identity.Account;
     core.setOutput('aws-account-id', accountId);
diff --git a/index.js b/index.js
index 48e0165..fb3a1a4 100644
--- a/index.js
+++ b/index.js
@@ -32,7 +32,9 @@ async function run() {
     core.exportVariable('AWS_REGION', region);
 
     // Get the AWS account ID
-    const sts = new aws.STS();
+    const sts = new aws.STS({
+      customUserAgent: 'configure-aws-credentials-for-github-actions'
+    });
     const identity = await sts.getCallerIdentity().promise();
     const accountId = identity.Account;
     core.setOutput('aws-account-id', accountId);