diff --git a/ChatBot.Tests/Services/OllamaClientAdapterTests.cs b/ChatBot.Tests/Services/OllamaClientAdapterTests.cs
index 79feee0..a6bf7cd 100644
--- a/ChatBot.Tests/Services/OllamaClientAdapterTests.cs
+++ b/ChatBot.Tests/Services/OllamaClientAdapterTests.cs
@@ -131,7 +131,7 @@ public class OllamaClientAdapterTests
 
         // Assert
         Assert.NotNull(result);
-        Assert.IsAssignableFrom<IAsyncEnumerable<ChatResponseStream?>>(result);
+        Assert.IsType<IAsyncEnumerable<ChatResponseStream?>>(result, exactMatch: false);
     }
 
     [Fact]
@@ -146,7 +146,7 @@ public class OllamaClientAdapterTests
 
         // Assert
         Assert.NotNull(result);
-        Assert.IsAssignableFrom<Task<IEnumerable<Model>>>(result);
+        Assert.IsType<Task<IEnumerable<Model>>>(result, exactMatch: false);
     }
 
     [Fact]