diff --git a/langfun/core/llms/vertexai_test.py b/langfun/core/llms/vertexai_test.py
index 57b9358..9295868 100644
--- a/langfun/core/llms/vertexai_test.py
+++ b/langfun/core/llms/vertexai_test.py
@@ -195,8 +195,12 @@ def test_generation_config(self):
             stop=['\n'],
         ),
     )
+    actual = config.to_dict()
+    # There is a discrepancy between the `property_ordering` in the
+    # Google-internal version and the open-source version.
+    actual['response_schema'].pop('property_ordering', None)
     self.assertEqual(
-        config.to_dict(),
+        actual,
         dict(
             temperature=2.0,
             top_p=1.0,
@@ -209,7 +213,6 @@ def test_generation_config(self):
                 'properties': {
                     'name': {'type_': 'STRING'}
                 },
-                'property_ordering': ['name'],
                 'required': ['name'],
                 'title': 'Person',
             }