diff --git a/python/pyspark/sql/column.py b/python/pyspark/sql/column.py index 8dc5039f587f..f94501e2de23 100644 --- a/python/pyspark/sql/column.py +++ b/python/pyspark/sql/column.py @@ -333,7 +333,7 @@ def when(self, condition, value): if not isinstance(condition, Column): raise TypeError("condition should be a Column") v = value._jc if isinstance(value, Column) else value - jc = sc._jvm.functions.when(condition._jc, v) + jc = self._jc.when(condition._jc, v) return Column(jc) @since(1.4)