diff --git a/pyrms/rmsTest.py b/pyrms/rmsTest.py
index 87777b5..5baf7af 100644
--- a/pyrms/rmsTest.py
+++ b/pyrms/rmsTest.py
@@ -15,4 +15,4 @@ def test_simulate(self):
         sol = rms.solve(react.ode,rms.CVODE_BDF(),abstol=1e-20,reltol=1e-8)
         sim = rms.Simulation(sol,domain)
         self.assertAlmostEqual(rms.molefractions(sim,"H2",3.0),0.34445669,places=3)
-        rms.pygetfluxdiagram(sim,1.0)
+        rms.getfluxdiagram(sim,1.0)