From 1529e43c68fabcfb72832ebd6b04523746c67dbd Mon Sep 17 00:00:00 2001
From: ddivad195 <ddivad195@github.com>
Date: Tue, 5 Mar 2024 16:31:13 +0000
Subject: [PATCH] fix isEnterprise check

---
 src/config.ts | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/config.ts b/src/config.ts
index 2b88d94..0115080 100644
--- a/src/config.ts
+++ b/src/config.ts
@@ -93,10 +93,9 @@ export async function resolvePublishActionOptions(): Promise<PublishActionOption
   const containerRegistryUrl: URL =
     await iaToolkit.getContainerRegistryURL(apiBaseUrl)
 
-  // TODO: Figure out if there's a better way to do this
   const isEnterprise =
-    !githubServerUrl.endsWith('github.com') &&
-    !githubServerUrl.endsWith('ghe.com')
+    !githubServerUrl.endsWith('.github.com') &&
+    !githubServerUrl.endsWith('.ghe.com')
 
   return {
     event,