diff --git a/tests/test_basic.py b/tests/test_basic.py
index 9fe4256..d3a103a 100644
--- a/tests/test_basic.py
+++ b/tests/test_basic.py
@@ -16,9 +16,9 @@ def test_basic():
 
 
 def test_coords():
-    with xr.open_dataset(EXAMPLE_FILES_DIR / "0011.sdf") as df:
-        px_photon = "dist_fn/px_py/Photon"
-        assert px_photon in df
-        x_coord = "Px_px_py/Photon"
-        assert x_coord in df[px_photon].coords
+    with xr.open_dataset(EXAMPLE_FILES_DIR / "0010.sdf") as df:
+        px_electron = "dist_fn/x_px/electron"
+        assert px_electron in df
+        x_coord = "Px_x_px/electron"
+        assert x_coord in df[px_electron].coords
         assert df[x_coord].attrs["long_name"] == "Px"