diff --git a/scripts/gen_openapi.sh b/scripts/gen_openapi.sh
index fa5ca90733..7db57ef24e 100755
--- a/scripts/gen_openapi.sh
+++ b/scripts/gen_openapi.sh
@@ -89,7 +89,8 @@ vault secrets enable "totp"
 vault secrets enable "transit"
 
 # Enable enterprise features
-if [[ -n "${VAULT_LICENSE:-}" ]]; then
+# Check if vault version contains +ent
+if vault version | grep -q "+ent"; then
     vault secrets enable "keymgmt"
     vault secrets enable "kmip"
     vault secrets enable "transform"