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 e7e990992f4..04cc69231fe 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
@@ -116,7 +116,7 @@ def test_catboost_with_categorical_features():
             pd.DataFrame(rng.standard_normal((100, 5))),
             pd.Series(rng.standard_normal(100)),
         ),
-        (rng.standard_normal(100, 5), rng.standard_normal(100)),
+        (rng.standard_normal((100, 5)), rng.standard_normal(100)),
     ],
 )
 def test_catboost_train_test_split(X, y):