diff --git a/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs b/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs
index d306966..b729374 100644
--- a/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs
+++ b/tests/Mscc.GenerativeAI/GoogleAi_Embedding_Should.cs
@@ -22,8 +22,8 @@ public GoogleAi_Embedding_Should(ITestOutputHelper output, ConfigurationFixture
         public void Initialize_Model()
         {
             // Arrange
-            var expected = Model.Embedding;
-            
+            var expected = Model.Embedding.SanitizeModelName();
+
             // Act
             var model = new GenerativeModel(apiKey: fixture.ApiKey, model: this.model);