diff --git a/pyrms/rmsTest.py b/pyrms/rmsTest.py
index b0b7b91..87777b5 100644
--- a/pyrms/rmsTest.py
+++ b/pyrms/rmsTest.py
@@ -15,3 +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)