diff --git a/Package.swift b/Package.swift
index 1066b85..f427e6b 100644
--- a/Package.swift
+++ b/Package.swift
@@ -19,7 +19,7 @@ let package = Package(
         )
     ],
     dependencies: [
-        .package(url: "https://github.com/andrea-scuderi/swift-aws-lambda-runtime.git", branch: "main"),
+        .package(url: "https://github.com/swift-server/swift-aws-lambda-runtime.git", from: "2.0.0-beta.1"),
         .package(url: "https://github.com/swift-server/swift-aws-lambda-events.git", from: "0.5.0"),
         .package(url: "https://github.com/swift-server/async-http-client.git", from: "1.22.0"),
         .package(url: "https://github.com/swift-server/swift-service-lifecycle.git", from: "2.6.3"),
diff --git a/Tests/BreezeLambdaWebHookTests/Lambda.swift b/Tests/BreezeLambdaWebHookTests/Lambda.swift
index fbd1942..1b67054 100644
--- a/Tests/BreezeLambdaWebHookTests/Lambda.swift
+++ b/Tests/BreezeLambdaWebHookTests/Lambda.swift
@@ -85,7 +85,7 @@ final actor MockLambdaResponseStreamWriter: LambdaResponseStreamWriter {
         self.buffer = buffer
     }
     
-    func write(_ buffer: ByteBuffer) async throws {
+    func write(_ buffer: ByteBuffer, hasCustomHeaders: Bool) async throws {
         fatalError("Unexpected call")
     }