diff --git a/test/Amazon.Lambda.Tools.Test/ApplySettingsTest.cs b/test/Amazon.Lambda.Tools.Test/ApplySettingsTest.cs
index 8cbe4ed7..0a557826 100644
--- a/test/Amazon.Lambda.Tools.Test/ApplySettingsTest.cs
+++ b/test/Amazon.Lambda.Tools.Test/ApplySettingsTest.cs
@@ -109,7 +109,7 @@ public async Task SetLoggingPropertiesForUpdateRequest()
                     Assert.Equal("DEBUG", request.LoggingConfig.ApplicationLogLevel);
                     Assert.Equal("WARN", request.LoggingConfig.SystemLogLevel);
                 })
-                .Returns((CreateFunctionRequest r, CancellationToken token) =>
+                .Returns((UpdateFunctionConfigurationRequest r, CancellationToken token) =>
                 {
                     return Task.FromResult(new UpdateFunctionConfigurationResponse());
                 });