diff --git a/Common/version.props b/Common/version.props
index a7f29204a..82aa0c233 100644
--- a/Common/version.props
+++ b/Common/version.props
@@ -2,7 +2,7 @@
 <!-- This file may be overwritten by automation. Only values allowed here are VersionPrefix and VersionSuffix. -->
 <Project>
   <PropertyGroup>
-    <VersionPrefix>1.7.11</VersionPrefix>
+    <VersionPrefix>1.7.12</VersionPrefix>
     <VersionSuffix></VersionSuffix>
   </PropertyGroup>
 </Project>
diff --git a/Source/Core/Runtime/CoyoteRuntime.cs b/Source/Core/Runtime/CoyoteRuntime.cs
index 3b49da68f..2d9ee4073 100644
--- a/Source/Core/Runtime/CoyoteRuntime.cs
+++ b/Source/Core/Runtime/CoyoteRuntime.cs
@@ -361,7 +361,15 @@ internal Task RunTestAsync(Delegate testMethod, string testName)
             Action runTest = () =>
             {
                 Task task = Task.CompletedTask;
-                if (this.Extension.RunTest(testMethod, out Task extensionTask))
+                if (testMethod is Action<ICoyoteRuntime> actionWithRuntime)
+                {
+                    actionWithRuntime(this);
+                }
+                else if (testMethod is Func<ICoyoteRuntime, Task> functionWithRuntime)
+                {
+                    task = functionWithRuntime(this);
+                }
+                else if (this.Extension.RunTest(testMethod, out Task extensionTask))
                 {
                     task = extensionTask;
                 }