diff --git a/SpeckleAutomateDotnetExample/SpeckleAutomateDotnetExample.csproj b/SpeckleAutomateDotnetExample/SpeckleAutomateDotnetExample.csproj
index c883e57..f1b4a10 100644
--- a/SpeckleAutomateDotnetExample/SpeckleAutomateDotnetExample.csproj
+++ b/SpeckleAutomateDotnetExample/SpeckleAutomateDotnetExample.csproj
@@ -8,8 +8,8 @@
   </PropertyGroup>
 
   <ItemGroup>
-    <PackageReference Include="Speckle.Automate.Sdk" Version="2.19.2" />
-    <PackageReference Include="Speckle.Objects" Version="2.19.2" />
+    <PackageReference Include="Speckle.Automate.Sdk" Version="2.21.0" />
+    <PackageReference Include="Speckle.Objects" Version="2.21.0" />
   </ItemGroup>
 
 </Project>
diff --git a/TestAutomateFunction/AutomationContextTest.cs b/TestAutomateFunction/AutomationContextTest.cs
index beea937..692a691 100644
--- a/TestAutomateFunction/AutomationContextTest.cs
+++ b/TestAutomateFunction/AutomationContextTest.cs
@@ -3,6 +3,7 @@ namespace TestAutomateFunction;
 using Speckle.Automate.Sdk;
 using Speckle.Automate.Sdk.Test;
 using Speckle.Core.Api;
+using Speckle.Core.Api.GraphQL.Models;
 using Speckle.Core.Credentials;
 
 [TestFixture]