diff --git a/src/async-function.ts b/src/async-function.ts
index 0356b6d..8b72c82 100644
--- a/src/async-function.ts
+++ b/src/async-function.ts
@@ -1,16 +1,16 @@
 import * as core from '@actions/core'
 import * as exec from '@actions/exec'
-import {context as _ghContext, GitHub} from '@actions/github/lib/utils'
+import type {context, getOctokit} from '@actions/github'
 import * as glob from '@actions/glob'
 import * as io from '@actions/io'
 
 const AsyncFunction = Object.getPrototypeOf(async () => null).constructor
 
 export declare type AsyncFunctionArguments = {
-  context: typeof _ghContext
+  context: typeof context
   core: typeof core
-  github: InstanceType<typeof GitHub>
-  octokit: InstanceType<typeof GitHub>
+  github: ReturnType<typeof getOctokit>
+  octokit: ReturnType<typeof getOctokit>
   exec: typeof exec
   glob: typeof glob
   io: typeof io