diff --git a/pandas-stubs/core/frame.pyi b/pandas-stubs/core/frame.pyi
index 87b4a55e..0c8166f0 100644
--- a/pandas-stubs/core/frame.pyi
+++ b/pandas-stubs/core/frame.pyi
@@ -2272,7 +2272,7 @@ class DataFrame(NDFrame, OpsMixin, _GetItemHack):
         replace: _bool = ...,
         weights: _str | ListLike | None = ...,
         random_state: RandomState | None = ...,
-        axis: AxisIndex | None = ...,
+        axis: Axis | None = ...,
         ignore_index: _bool = ...,
     ) -> Self: ...
     def sem(
diff --git a/tests/test_frame.py b/tests/test_frame.py
index eb0da931..9c7599c8 100644
--- a/tests/test_frame.py
+++ b/tests/test_frame.py
@@ -351,6 +351,7 @@ def test_types_sample() -> None:
         assert_type(df.sample(n=1, random_state=np.random.default_rng()), pd.DataFrame),
         pd.DataFrame,
     )
+    check(assert_type(df.sample(n=1, axis=1), pd.DataFrame), pd.DataFrame)
 
 
 def test_types_nlargest_nsmallest() -> None: