diff --git a/CADRE/test/test_derivatives.py b/CADRE/test/test_derivatives.py
index 04c65b4..98447e2 100644
--- a/CADRE/test/test_derivatives.py
+++ b/CADRE/test/test_derivatives.py
@@ -66,7 +66,7 @@ def setup(self, compname, inputs, state0):
         self.prob.model.add_subsystem('indep', indep, promotes=['*'])
         self.prob.model.add_subsystem('comp', comp, promotes=['*'])
 
-        self.prob.setup()
+        self.prob.setup(mode='fwd')
 
     def compare_derivatives(self, var_in, var_out, rel_error=False):
         # check totals
diff --git a/CADRE/test/test_rk_deriv.py b/CADRE/test/test_rk_deriv.py
index f66d7ce..8b373ab 100644
--- a/CADRE/test/test_rk_deriv.py
+++ b/CADRE/test/test_rk_deriv.py
@@ -121,7 +121,7 @@ def test_rk_with_time_invariant(self):
         prob.model.add_subsystem('indeps', indeps, promotes=['*'])
         prob.model.add_subsystem('rktest', rktest, promotes=['*'])
 
-        prob.setup()
+        prob.setup(mode='fwd')
         prob.run_model()
 
         # check partials