diff --git a/tests/test_core.py b/tests/test_core.py
index 1012b69..592e733 100644
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -252,9 +252,14 @@ def test_nan_inf_attributes():
             X2 = client["X"]
             assert isinstance(X2, lindi.LindiH5pyDataset)
 
-            assert X2.attrs["nan"] == "NaN"
-            assert X2.attrs["inf"] == "Infinity"
-            assert X2.attrs["ninf"] == "-Infinity"
+            nanval = X1.attrs["nan"]
+            assert isinstance(nanval, float) and np.isnan(nanval)
+            assert X1.attrs["inf"] == np.inf
+            assert X1.attrs["ninf"] == -np.inf
+
+            # assert X2.attrs["nan"] == "NaN"
+            # assert X2.attrs["inf"] == "Infinity"
+            # assert X2.attrs["ninf"] == "-Infinity"
 
 
 def test_reference_file_system_to_file():