diff --git a/synbio_morpher/utils/circuit/agnostic_circuits/circuit_manager.py b/synbio_morpher/utils/circuit/agnostic_circuits/circuit_manager.py
index 7b6a238b..86130048 100644
--- a/synbio_morpher/utils/circuit/agnostic_circuits/circuit_manager.py
+++ b/synbio_morpher/utils/circuit/agnostic_circuits/circuit_manager.py
@@ -269,7 +269,7 @@ def compute_steady_states(self, circuits: List[Circuit],
                 b_copynumbers.append(copynumbers)
                 t = steady_state_result.t
 
-        elif solver_type == 'jax':
+        elif solver_type == 'diffrax':
             ref_circuit = circuits[0]
             forward_rates = ref_circuit.qreactions.reactions.forward_rates
             reverse_rates = np.asarray(