diff --git a/bluemira/codes/openmc/solver.py b/bluemira/codes/openmc/solver.py
index f91b1e7abc..eca8f5fe30 100644
--- a/bluemira/codes/openmc/solver.py
+++ b/bluemira/codes/openmc/solver.py
@@ -353,7 +353,7 @@ def run(
         """Run stage for Teardown task"""
         result = OpenMCResult.from_run(
             universe,
-            n_DT_reactions(source_params.plasma_physics_units.reactor_power),
+            n_DT_reactions(source_params.reactor_power),
             statepoint_file,
         )
         if delete_files:
diff --git a/bluemira/fuel_cycle/tools.py b/bluemira/fuel_cycle/tools.py
index 7ff3a3f0ed..33a5408981 100644
--- a/bluemira/fuel_cycle/tools.py
+++ b/bluemira/fuel_cycle/tools.py
@@ -482,7 +482,7 @@ def legal_limit(
         mb = None
 
     if mb is None:
-        mb = raw_uc(r_T_burn(p_fus), "kg/s", "g/s")
+        mb = r_T_burn(p_fus)
 
     m_plasma = (
         (mb * ((1 / fb - 1) + (1 - eta_fuel_pump) * (1 - eta_f) / (eta_f * fb)) + m_gas)