diff --git a/src/Speckle.Automate.Sdk/AutomationContext.cs b/src/Speckle.Automate.Sdk/AutomationContext.cs
index b3b8a638..7bd0cc1e 100644
--- a/src/Speckle.Automate.Sdk/AutomationContext.cs
+++ b/src/Speckle.Automate.Sdk/AutomationContext.cs
@@ -63,7 +63,7 @@ public async Task<Base> ReceiveVersion(CancellationToken cancellationToken = def
       );
     }
 
-    Base? rootObject = await operations
+    Base rootObject = await operations
       .Receive2(
         SpeckleClient.ServerUrl,
         AutomationRunData.ProjectId,
@@ -74,6 +74,10 @@ public async Task<Base> ReceiveVersion(CancellationToken cancellationToken = def
       )
       .ConfigureAwait(false);
 
+    await SpeckleClient
+      .Version.Received(new(version.id, AutomationRunData.ProjectId, "automate_function"), cancellationToken)
+      .ConfigureAwait(false);
+
     Console.WriteLine($"It took {Elapsed.TotalSeconds} seconds to receive the speckle version {versionId}");
     return rootObject;
   }