diff --git a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py
index 922f3f385df..e7e990992f4 100644
--- a/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py
+++ b/python/cudf/cudf_pandas_tests/third_party_integration_tests/tests/test_catboost.py
@@ -113,7 +113,7 @@ def test_catboost_with_categorical_features():
     "X, y",
     [
         (
-            pd.DataFrame(rng.standard_normal(100, 5)),
+            pd.DataFrame(rng.standard_normal((100, 5))),
             pd.Series(rng.standard_normal(100)),
         ),
         (rng.standard_normal(100, 5), rng.standard_normal(100)),