diff --git a/tasks/install_pe.json b/tasks/install_pe.json
index c17ffe8..7c7ca3a 100644
--- a/tasks/install_pe.json
+++ b/tasks/install_pe.json
@@ -6,6 +6,10 @@
       "version": {
       "description": "The release of PE you want to install e.g. 2018.1 (Default: 2019.2)",
       "type": "Optional[String[1]]"
+    },
+      "os": {
+      "description": "The os of PE you are installing e.g. ubuntu-22.04-amd64  (Default: el-7-x86_64)",
+      "type": "Optional[String[1]]"
     }
   },
   "private": true
diff --git a/tasks/install_pe.sh b/tasks/install_pe.sh
index 832c1a7..24e98e7 100644
--- a/tasks/install_pe.sh
+++ b/tasks/install_pe.sh
@@ -7,8 +7,15 @@ else
   PE_RELEASE=$PT_version
 fi
 
+if [ -z ${PT_os+x} ]; then
+  PE_OS=el-7-x86_64
+
+else
+  PE_OS=$PT_os
+fi
+
 PE_LATEST=$(curl https://artifactory.delivery.puppetlabs.net/artifactory/generic_enterprise__local/"${PE_RELEASE}"/ci-ready/LATEST)
-PE_FILE_NAME=puppet-enterprise-${PE_LATEST}-el-7-x86_64
+PE_FILE_NAME=puppet-enterprise-${PE_LATEST}-${PE_OS}
 TAR_FILE=${PE_FILE_NAME}.tar
 DOWNLOAD_URL=https://artifactory.delivery.puppetlabs.net/artifactory/generic_enterprise__local/${PE_RELEASE}/ci-ready/${TAR_FILE}