diff --git a/dm_pix/_src/color_conversion_test.py b/dm_pix/_src/color_conversion_test.py
index 282e123..bd36749 100644
--- a/dm_pix/_src/color_conversion_test.py
+++ b/dm_pix/_src/color_conversion_test.py
@@ -358,8 +358,8 @@ def test_mixed_image_values_and_gradients(self):
   @chex.all_variants
   def test_chromatic_gradients_match_finite_differences(self):
     x = jnp.array([0.2, 0.6, 0.9])
+    x64 = np.asarray(x, dtype=np.float64)
     numerical = np.zeros((3, 3))
-    x64 = np.asarray(x).astype(np.float64)
     for index in range(3):
       delta = np.zeros(3)
       delta[index] = 1e-5