Fix test variable name

AlexHls · Nov 1, 2023 · a66bf8e · a66bf8e
1 parent 8eaf6bb
commit a66bf8e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tardis/plasma/tests/test_nlte_solver.py b/tardis/plasma/tests/test_nlte_solver.py
@@ -260,7 +260,7 @@ def nlte_raw_plasma_w0(
 
 def test_critical_case_w1(nlte_raw_plasma_w1):
     """Check that the LTE and NLTE solution agree for w=1.0."""
-    ion_number_density_nlte = nlte_raw_plasma_w1.ion_number_density_nlte.values
+    ion_number_density_nlte = nlte_raw_plasma_w1.ion_number_density.values
     ion_number_density_nlte[ion_number_density_nlte < 1e-10] = 0.0
 
     ind = IonNumberDensity(nlte_raw_plasma_w1)