diff --git a/tests/GeminiDotnet.Integration.SemanticKernel.Tests/GeminiDotnet.Integration.SemanticKernel.Tests.csproj b/tests/GeminiDotnet.Integration.SemanticKernel.Tests/GeminiDotnet.Integration.SemanticKernel.Tests.csproj
index 3a45a33..7f0c039 100644
--- a/tests/GeminiDotnet.Integration.SemanticKernel.Tests/GeminiDotnet.Integration.SemanticKernel.Tests.csproj
+++ b/tests/GeminiDotnet.Integration.SemanticKernel.Tests/GeminiDotnet.Integration.SemanticKernel.Tests.csproj
@@ -25,6 +25,7 @@
     </ItemGroup>
 
     <ItemGroup>
+      <PackageReference Include="Microsoft.Extensions.AI" />
         <ProjectReference Include="..\..\src\GeminiDotnet.Extensions.AI\GeminiDotnet.Extensions.AI.csproj" />
     </ItemGroup>