diff --git a/test/SourceGeneratorTests/Baselines/MainTests/TaskOfTMethod/Pretender_g.cs b/test/SourceGeneratorTests/Baselines/MainTests/TaskOfTMethod/Pretender_g.cs
index f3f5a67..aef1da2 100644
--- a/test/SourceGeneratorTests/Baselines/MainTests/TaskOfTMethod/Pretender_g.cs
+++ b/test/SourceGeneratorTests/Baselines/MainTests/TaskOfTMethod/Pretender_g.cs
@@ -59,11 +59,13 @@ file static class SetupInterceptors
             {
                 Matcher matchCall = (callInfo, setup) =>
                 {
-                    var singleUseCallHandler = new SingleUseCallHandler();
+                    var singleUseCallHandler = new SingleUseCallHandler<global::System.Threading.Tasks.Task<string>>();
                     var fake = new PretendIMyInterface(singleUseCallHandler);
+
                     var listener = MatcherListener.StartListening();
                     setup.Method.Invoke(setup.Target, [fake]);
                     listener.Dispose();
+
                     var capturedArguments = singleUseCallHandler.Arguments;
 
                     var str_capturedMatcher = listener.Matchers[0];