diff --git a/src/action.js b/src/action.js
index 9070842..30b23ec 100644
--- a/src/action.js
+++ b/src/action.js
@@ -47,7 +47,7 @@ const getRangeForPushEvent = () => {
 const getRangeForEvent = async () => {
   if (GITHUB_EVENT_NAME !== pullRequestEvent) return getRangeForPushEvent()
 
-  const octokit = new github.GitHub(core.getInput('token'))
+  const octokit = github.getOctokit(core.getInput('token'))
   const { owner, repo, number } = eventContext.issue
   const { data: commits } = await octokit.pulls.listCommits({
     owner,
diff --git a/src/action.test.js b/src/action.test.js
index 7810dd7..a4a2dd5 100644
--- a/src/action.test.js
+++ b/src/action.test.js
@@ -32,7 +32,7 @@ const runAction = () => {
   }
 
   updateEnvVars({ GITHBU_TOKEN: 'test-github-token' })
-  td.replace(github, 'GitHub', MockOctokit)
+  td.replace(github, 'getOctokit', () => new MockOctokit())
 
   return require('./action')()
 }