diff --git a/__test__/github-api-helper.test.ts b/__test__/github-api-helper.test.ts
new file mode 100644
index 0000000..0e8920b
--- /dev/null
+++ b/__test__/github-api-helper.test.ts
@@ -0,0 +1,37 @@
+import * as github from '@actions/github'
+import * as githubApiHelper from '../lib/github-api-helper'
+
+jest.mock('@actions/github')
+
+describe('github-api-helper tests', () => {
+  describe('github enterprise compatibility', () => {
+    beforeEach(() => {
+      process.env.GITHUB_SERVER_URL = 'https://enterprise.git.com'
+    })
+
+    afterEach(() => {
+      delete process.env.GITHUB_SERVER_URL
+    })
+
+    it('getDefaultBranch should use GITHUB_SERVER_URL to set the baseUrl', async () => {
+      ;(github.getOctokit as jest.Mock).mockImplementation(() => {
+        return {
+          rest: {
+            repos: {
+              get: jest.fn(() => ({data: {default_branch: 'default-branch'}}))
+            }
+          }
+        }
+      })
+
+      await githubApiHelper.getDefaultBranch('token', 'owner', 'repo')
+
+      expect(github.getOctokit).toHaveBeenCalledWith(
+        'token',
+        expect.objectContaining({
+          baseUrl: 'https://enterprise.git.com/api/v3'
+        })
+      )
+    })
+  })
+})
diff --git a/dist/index.js b/dist/index.js
index 1389602..b64f3b6 100644
--- a/dist/index.js
+++ b/dist/index.js
@@ -1526,6 +1526,7 @@ const io = __importStar(__nccwpck_require__(7436));
 const path = __importStar(__nccwpck_require__(1017));
 const retryHelper = __importStar(__nccwpck_require__(2155));
 const toolCache = __importStar(__nccwpck_require__(7784));
+const url_helper_1 = __nccwpck_require__(9437);
 const v4_1 = __importDefault(__nccwpck_require__(824));
 const url_helper_1 = __nccwpck_require__(9437);
 const IS_WINDOWS = process.platform === 'win32';
diff --git a/src/github-api-helper.ts b/src/github-api-helper.ts
index b6bfc13..11b30a0 100644
--- a/src/github-api-helper.ts
+++ b/src/github-api-helper.ts
@@ -6,6 +6,7 @@ import * as io from '@actions/io'
 import * as path from 'path'
 import * as retryHelper from './retry-helper'
 import * as toolCache from '@actions/tool-cache'
+import {getServerApiUrl} from './url-helper'
 import {default as uuid} from 'uuid/v4'
 import {getServerApiUrl} from './url-helper'