diff --git a/python/pyspark/sql/connect/dataframe.py b/python/pyspark/sql/connect/dataframe.py index bf9ed83615b69..03a766aff3085 100644 --- a/python/pyspark/sql/connect/dataframe.py +++ b/python/pyspark/sql/connect/dataframe.py @@ -215,8 +215,7 @@ def groupBy(self, *cols: "ColumnOrString") -> GroupingFrame: return GroupingFrame(self, *cols) def head(self, n: int) -> Optional["pandas.DataFrame"]: - self.limit(n) - return self.toPandas() + return self.limit(n).toPandas() # TODO(martin.grund) fix mypu def join(self, other: "DataFrame", on: Any, how: Optional[str] = None) -> "DataFrame":