diff --git a/THIRD-PARTY-LICENSES.md b/THIRD-PARTY-LICENSES.md
index 6852840..5eb5dfc 100644
--- a/THIRD-PARTY-LICENSES.md
+++ b/THIRD-PARTY-LICENSES.md
@@ -57,8 +57,8 @@ CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
 ## github.com/go-chi/chi/v5
 
 * Name: github.com/go-chi/chi/v5
-* Version: v5.2.2
-* License: [MIT](https://github.com/go-chi/chi/blob/v5.2.2/LICENSE)
+* Version: v5.4.0
+* License: [MIT](https://github.com/go-chi/chi/blob/v5.4.0/LICENSE)
 
 ```
 Copyright (c) 2015-present Peter Kieltyka (https://github.com/pkieltyka), Google Inc.
diff --git a/internal/lambda-managed-instances/invoke/running_invoke_test.go b/internal/lambda-managed-instances/invoke/running_invoke_test.go
index cf68771..6a64e6b 100644
--- a/internal/lambda-managed-instances/invoke/running_invoke_test.go
+++ b/internal/lambda-managed-instances/invoke/running_invoke_test.go
@@ -98,7 +98,7 @@ func mockMetricsBeforeResponse(mocks *runningInvokeMocks) {
 	mocks.metrics.On("TriggerStartRequest")
 	mocks.metrics.On("SendInvokeStartEvent", mock.AnythingOfType("*interop.TracingCtx")).Return(nil)
 	mocks.metrics.On("TriggerSentRequest", mock.AnythingOfType("int64"), mock.AnythingOfType("time.Duration"), mock.AnythingOfType("time.Duration")).Return()
-	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.AnythingOfType("json.RawMessage")).Return(nil)
+	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.IsType(json.RawMessage{})).Return(nil)
 }
 
 func checkRunningInvokeMockExpectations(t *testing.T, mocks *runningInvokeMocks) {
@@ -250,7 +250,7 @@ func TestRuntimeErrorFailure_SendInvokeToRuntime_Error(t *testing.T) {
 	mocks.metrics.On("TriggerStartRequest")
 	mocks.metrics.On("SendInvokeStartEvent", mock.AnythingOfType("*interop.TracingCtx")).Return(nil)
 	mocks.metrics.On("TriggerSentResponse", false, err, mock.Anything, 0).Return()
-	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.AnythingOfType("json.RawMessage")).Return(nil)
+	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.IsType(json.RawMessage{})).Return(nil)
 
 	invokeErr := runInvoke.RunInvokeAndSendResult(mocks.ctx, &mocks.staticData, &mocks.eaInvokeRequest, &mocks.metrics, &mocks.eaInvokeResponder)
 	assert.Error(t, invokeErr)
@@ -282,7 +282,7 @@ func TestRuntimeErrorFailure_SendInvokeToRuntime_Timeout(t *testing.T) {
 	mocks.metrics.On("TriggerStartRequest")
 	mocks.metrics.On("SendInvokeStartEvent", mock.AnythingOfType("*interop.TracingCtx")).Return(nil)
 	mocks.metrics.On("TriggerSentResponse", false, err, mock.Anything, 0).Return()
-	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.AnythingOfType("json.RawMessage")).Return(nil)
+	mocks.metrics.On("SendInvokeFinishedEvent", mock.AnythingOfType("*interop.TracingCtx"), mock.IsType(json.RawMessage{})).Return(nil)
 
 	invokeErr := runInvoke.RunInvokeAndSendResult(mocks.ctx, &mocks.staticData, &mocks.eaInvokeRequest, &mocks.metrics, &mocks.eaInvokeResponder)
 	assert.Error(t, invokeErr)