diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml
index eaa3ab5..2c712b2 100644
--- a/.github/workflows/test.yml
+++ b/.github/workflows/test.yml
@@ -1,6 +1,10 @@
 name: Test
-
-on: [push]
+on:
+  push:
+    branches:
+      - main
+  pull_request:
+    types: [opened, synchronize]
 
 jobs:
   test_matrix:
diff --git a/index.d.ts b/index.d.ts
index c3d04d1..2dc6ca8 100644
--- a/index.d.ts
+++ b/index.d.ts
@@ -1,5 +1,9 @@
 import { Probot } from "probot";
-import { APIGatewayProxyHandler } from "aws-lambda";
+import {
+  APIGatewayProxyEvent,
+  APIGatewayProxyResult,
+  Context,
+} from "aws-lambda";
 import { ApplicationFunction } from "probot/lib/types";
 
 export * from "probot";
@@ -7,4 +11,7 @@ export * from "probot";
 export function createLambdaFunction(
   app: ApplicationFunction,
   options: { probot: Probot }
-): APIGatewayProxyHandler;
+): (
+  event: APIGatewayProxyEvent,
+  context: Context
+) => Promise<APIGatewayProxyResult>;