diff --git a/test/algorithm_test/XYZAlgorithm_test.py b/test/algorithm_test/XYZAlgorithm_test.py
index 0102f90688f202cf664b8b4c2049b6fd78263362..8d08574477dc5241c4524ac8e04ce7d67f2d6601 100644
--- a/test/algorithm_test/XYZAlgorithm_test.py
+++ b/test/algorithm_test/XYZAlgorithm_test.py
@@ -54,7 +54,7 @@ def test_xyzalgorithm_limited_channels():
     # d has `count` values (same as input)
     assert_equals(len(d.data), count)
     # d has no NaN values
-    assert_equals(len(d[d == np.NaN]), 0)
+    assert_equals(np.isnan(d).any(), False)
 
 
 def test_xyzalgorithm_uneccesary_channel_empty():