diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index a0073d0..3b712f1 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -7,7 +7,7 @@ jobs:
   test:
     strategy:
       matrix:
-        python-version: [3.11]
+        python-version: ["3.9", "3.11", "3.12"]
 
     runs-on: ubuntu-latest
 
diff --git a/tests/test_ndarray.py b/tests/test_ndarray.py
index b8697ed..0305371 100644
--- a/tests/test_ndarray.py
+++ b/tests/test_ndarray.py
@@ -96,7 +96,7 @@ def test_ndarray_serialize():
 
     # but when we just dump to a dict we don't coerce
     mod_dict = mod.model_dump()
-    assert isinstance(mod_dict["large_array"], np.ndarray)
+    assert isinstance(mod_dict["array"], np.ndarray)
 
 
 # def test_ndarray_proxy(tmp_output_dir_func):