diff --git a/tests/unit/dag/ops/test_udf.py b/tests/unit/dag/ops/test_udf.py
index 6a52b554f..038575025 100644
--- a/tests/unit/dag/ops/test_udf.py
+++ b/tests/unit/dag/ops/test_udf.py
@@ -229,7 +229,7 @@ def test_udf_dtype_multi_op_propagation(cpu):
         {
             "a": np.arange(size),
             "b": np.random.choice(["apple", "banana", "orange"], size),
-            "c": np.random.choice([0, 1], size).astype(np.float16),
+            "c": np.random.choice([0, 1], size),
         }
     )
     ddf0 = dd.from_pandas(df0, npartitions=4)