diff --git a/python/pyspark/sql/dataframe.py b/python/pyspark/sql/dataframe.py
index 3266c4135a764..042f3d1cba370 100644
--- a/python/pyspark/sql/dataframe.py
+++ b/python/pyspark/sql/dataframe.py
@@ -589,6 +589,7 @@ def writeStream(self) -> DataStreamWriter:
 
         Examples
         --------
+        >>> import time
         >>> import tempfile
         >>> df = spark.readStream.format("rate").load()
         >>> type(df.writeStream)
@@ -596,8 +597,11 @@ def writeStream(self) -> DataStreamWriter:
 
         >>> with tempfile.TemporaryDirectory(prefix="writeStream") as d:
         ...     # Create a table with Rate source.
-        ...     df.writeStream.toTable(
+        ...     query = df.writeStream.toTable(
         ...         "my_table", checkpointLocation=d)
+        ...     print(type(query))
+        ...     time.sleep(3)
+        ...     query.stop()
         <...streaming.query.StreamingQuery object at 0x...>
         """
         return DataStreamWriter(self)