diff --git a/dist/index.js b/dist/index.js
index 273d85b..472a75c 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -128230,9 +128230,8 @@ async function resolvePublishActionOptions() {
     }
     // Required Values fetched from the GitHub API
     const containerRegistryUrl = 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');
+    const isEnterprise = !githubServerUrl.endsWith('.github.com') &&
+        !githubServerUrl.endsWith('.ghe.com');
     return {
         event,
         ref,