diff --git a/plio/io/tests/test_io_bae.py b/plio/io/tests/test_io_bae.py
index 0059a2ca437b4d0a4d789cb31dd2cc644cda4d14..2817c889616371f63ee6b78ad875a9631ceca86a 100644
--- a/plio/io/tests/test_io_bae.py
+++ b/plio/io/tests/test_io_bae.py
@@ -61,7 +61,7 @@ def test_write_ipf(ipf, file):
     test_arr = np.hstack(np.array(test_arr))
     test_arr = test_arr.reshape(-1, 12)
 
-    assert (truth_arr==test_arr).all()
+    assert (truth_arr == test_arr).all()
 
 @pytest.mark.parametrize('gpf, file', [(insight_gpf(), 'out.gpf')])
 def test_write_gpf(gpf, file):
@@ -85,7 +85,7 @@ def test_write_gpf(gpf, file):
     truth_arr = np.genfromtxt(gpf, skip_header=3)
     test_arr = np.genfromtxt(file, skip_header=3)
 
-    assert (truth_arr==test_arr).all()
+    np.testing.assert_array_almost_equal(truth_arr, test_arr)
 
     # np.testing.assert_array_almost_equal(truth_arr, test_arr)