diff --git a/python/cuml/tests/test_array.py b/python/cuml/tests/test_array.py
index c4c479506c..f64683717d 100644
--- a/python/cuml/tests/test_array.py
+++ b/python/cuml/tests/test_array.py
@@ -1,5 +1,5 @@
 #
-# Copyright (c) 2020-2023, NVIDIA CORPORATION.
+# Copyright (c) 2020-2024, NVIDIA CORPORATION.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -251,8 +251,8 @@ def test_get_set_item(inp, indices, mem_type):
         _assert_equal(inp_view, ary[indices])
 
         # Check equality after assigning to array slice.
-        ary[indices] = 1.0
-        inp[indices] = 1.0
+        ary[indices] = inp.dtype.type(1.0)
+        inp[indices] = inp.dtype.type(1.0)
 
         # We need to assume that inp is not a cudf.Series here, otherwise
         # ary.to_output("cupy") called by equal() will trigger a