diff --git a/python/cuml/tests/test_input_utils.py b/python/cuml/tests/test_input_utils.py
index a3aef28446..fbbcde2105 100644
--- a/python/cuml/tests/test_input_utils.py
+++ b/python/cuml/tests/test_input_utils.py
@@ -396,7 +396,7 @@ def get_input(
         result = cudf.DataFrame(rand_mat, index=index)
 
     if type == "cudf-series":
-        result = cudf.Series(rand_mat, index=index)
+        result = cudf.Series(rand_mat.reshape(nrows), index=index)
 
     if type == "pandas":
         result = pdDF(cp.asnumpy(rand_mat), index=index)