diff --git a/types/async-function.d.ts b/types/async-function.d.ts
index b204e36..772435e 100644
--- a/types/async-function.d.ts
+++ b/types/async-function.d.ts
@@ -1,15 +1,15 @@
 /// <reference types="node" />
 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';
 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;