diff --git a/src/async-function.ts b/src/async-function.ts
index 84035f2..30613f1 100644
--- a/src/async-function.ts
+++ b/src/async-function.ts
@@ -1,17 +1,17 @@
 import * as core from '@actions/core'
 import * as exec from '@actions/exec'
-import {Context} from '@actions/github/lib/context'
-import {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: Context
+  context: typeof context
   core: typeof core
-  github: InstanceType<typeof GitHub>
-  octokit: InstanceType<typeof GitHub>
+  github: ReturnType<typeof getOctokit>
+  octokit: ReturnType<typeof getOctokit>
+  getOctokit: typeof getOctokit
   exec: typeof exec
   glob: typeof glob
   io: typeof io