diff --git a/python/pyspark/sql/tests/connect/test_connect_column_expressions.py b/python/pyspark/sql/tests/connect/test_connect_column_expressions.py
index ca2cc216ff2f0..bf2ffdf1dc09b 100644
--- a/python/pyspark/sql/tests/connect/test_connect_column_expressions.py
+++ b/python/pyspark/sql/tests/connect/test_connect_column_expressions.py
@@ -77,7 +77,7 @@ def test_column_expressions(self):
             ProtoExpression.UnresolvedAttribute,
         )
         self.assertEqual(
-            mod_fun.unresolved_function.arguments[0].unresolved_attribute.parts, ["id"]
+            mod_fun.unresolved_function.arguments[0].unresolved_attribute.unparsed_identifier, "id"
         )