diff --git a/tests/test_gpu_aamp_stimp.py b/tests/test_gpu_aamp_stimp.py
index 88f5a656d..1950d901f 100644
--- a/tests/test_gpu_aamp_stimp.py
+++ b/tests/test_gpu_aamp_stimp.py
@@ -59,7 +59,7 @@ def test_gpu_aamp_stimp(T):
     naive.replace_inf(ref_PAN)
     naive.replace_inf(cmp_PAN)
 
-    npt.assert_almost_equal(ref_PAN, cmp_PAN)
+    npt.assert_allclose(cmp_PAN, ref_PAN, atol=1.5e-07)
 
     # Compare transformed pan
     cmp_pan = pan.PAN_
@@ -76,4 +76,4 @@ def test_gpu_aamp_stimp(T):
     naive.replace_inf(ref_pan)
     naive.replace_inf(cmp_pan)
 
-    npt.assert_almost_equal(ref_pan, cmp_pan)
+    npt.assert_allclose(cmp_pan, ref_pan, atol=1.5e-07)