diff --git a/internal/lambda-managed-instances/invoke/running_invoke_test.go b/internal/lambda-managed-instances/invoke/running_invoke_test.go
index 951f650e..e077b0c2 100644
--- a/internal/lambda-managed-instances/invoke/running_invoke_test.go
+++ b/internal/lambda-managed-instances/invoke/running_invoke_test.go
@@ -99,7 +99,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(nil))).Return(nil)
 }
 
 func checkRunningInvokeMockExpectations(t *testing.T, mocks *runningInvokeMocks) {
@@ -248,7 +248,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(nil))).Return(nil)
 
 	invokeErr := runInvoke.RunInvokeAndSendResult(mocks.ctx, &mocks.staticData, &mocks.eaInvokeRequest, &mocks.metrics)
 	assert.Error(t, invokeErr)
@@ -280,7 +280,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(nil))).Return(nil)
 
 	invokeErr := runInvoke.RunInvokeAndSendResult(mocks.ctx, &mocks.staticData, &mocks.eaInvokeRequest, &mocks.metrics)
 	assert.Error(t, invokeErr)