diff --git a/tests/test_celestial_galsim.py b/tests/test_celestial_galsim.py
index 0d3d0dd169..86176a22a5 100644
--- a/tests/test_celestial_galsim.py
+++ b/tests/test_celestial_galsim.py
@@ -127,7 +127,7 @@ def test_angle():
     assert_raises(TypeError,galsim.AngleUnit, 1, 3)
     assert_raises(TypeError,galsim.Angle, 3.4)
     assert_raises(TypeError,galsim.Angle, theta1, galsim.degrees)
-    assert_raises((ValueError, TypeError), galsim.Angle, 'spam')
+    assert_raises((ValueError, TypeError), galsim.Angle, 'spam', galsim.degrees)
     assert_raises(TypeError,galsim.Angle, 1, 3)
 
 
diff --git a/tests/test_draw.py b/tests/test_draw.py
index 662f2f3b1f..8d337a65ee 100644
--- a/tests/test_draw.py
+++ b/tests/test_draw.py
@@ -1576,12 +1576,8 @@ def test_np_fft():
 def round_cast(array, dt):
     # array.astype(dt) doesn't round to the nearest for integer types.
     # This rounds first if dt is integer and then casts.
-    if is_jax_galsim():
-        # NOTE JAX doesn't round to the nearest int when drawing
-        pass
-    else:
-        if dt(0.5) != 0.5:
-            array = np.around(array)
+    if dt(0.5) != 0.5:
+        array = np.around(array)
     return array.astype(dt)
 
 @timer