diff --git a/pandas/io/formats/style.py b/pandas/io/formats/style.py
index 808b6979b235e..4d68971bf0ef6 100644
--- a/pandas/io/formats/style.py
+++ b/pandas/io/formats/style.py
@@ -577,7 +577,7 @@ def apply(self, func, axis=0, subset=None, **kwargs):
         -----
         The output shape of ``func`` should match the input, i.e. if
         ``x`` is the input row, column, or table (depending on ``axis``),
-        then ``func(x.shape) == x.shape`` should be true.
+        then ``func(x).shape == x.shape`` should be true.
 
         This is similar to ``DataFrame.apply``, except that ``axis=None``
         applies the function to the entire DataFrame at once,