diff --git a/test/unit/test_visualization.py b/test/unit/test_visualization.py
index bd5a74898..d2d7a4e6e 100644
--- a/test/unit/test_visualization.py
+++ b/test/unit/test_visualization.py
@@ -15,7 +15,7 @@ def test_sem_x(self):
 
     def test_xray_x(self):
         xray = ps.visualization.xray(self.im)
-        assert np.sum(xray) == np.sum(~self.im)
+        assert np.min(xray) >= 0 and np.max(xray) <= 1
 
     def test_sem_y(self):
         sem = ps.visualization.sem(self.im, axis=1)