diff --git a/provider/index.ts b/provider/index.ts
index e6c4a05..64b69ed 100644
--- a/provider/index.ts
+++ b/provider/index.ts
@@ -23,7 +23,10 @@ const filterUndefined = (items: Array): Array => {
};
const getTemplateUrl = (version: string): string => {
- return `https://s3.amazonaws.com/solutions-reference/aws-waf-security-automations/${version}/aws-waf-security-automations.template`;
+ if (version.startsWith('v3.')) {
+ return `https://s3.amazonaws.com/solutions-reference/aws-waf-security-automations/${version}/aws-waf-security-automations.template`;
+ }
+ return `https://s3.amazonaws.com/solutions-reference/security-automations-for-aws-waf/${version}/aws-waf-security-automations.template`;
};
interface CloudformationParameter {