diff --git a/firmware/controllers/algo/engine2.cpp b/firmware/controllers/algo/engine2.cpp
index 57baade851..30a0b2317a 100644
--- a/firmware/controllers/algo/engine2.cpp
+++ b/firmware/controllers/algo/engine2.cpp
@@ -150,9 +150,7 @@ void EngineState::periodicFastCallback() {
 	float untrimmedInjectionMass = getInjectionMass(rpm) * engine->engineState.lua.fuelMult + engine->engineState.lua.fuelAdd;
 	auto clResult = fuelClosedLoopCorrection();
 
-	float fuelLoad = getFuelingLoad();
-
-	injectionStage2Fraction = getStage2InjectionFraction(rpm, fuelLoad);
+	injectionStage2Fraction = getStage2InjectionFraction(rpm, engine->fuelComputer.afrTableYAxis);
 	float stage2InjectionMass = untrimmedInjectionMass * injectionStage2Fraction;
 	float stage1InjectionMass = untrimmedInjectionMass - stage2InjectionMass;
 
@@ -163,6 +161,7 @@ void EngineState::periodicFastCallback() {
 		? engine->module<InjectorModelSecondary>()->getInjectionDuration(stage2InjectionMass)
 		: 0;
 
+	float fuelLoad = getFuelingLoad();
 	injectionOffset = getInjectionOffset(rpm, fuelLoad);
 	engine->lambdaMonitor.update(rpm, fuelLoad);
 
diff --git a/firmware/tunerstudio/tunerstudio.template.ini b/firmware/tunerstudio/tunerstudio.template.ini
index bbd5842bbb..c668ba6c0c 100644
--- a/firmware/tunerstudio/tunerstudio.template.ini
+++ b/firmware/tunerstudio/tunerstudio.template.ini
@@ -1339,7 +1339,7 @@ curve = rangeMatrix, "Range Switch Input Matrix"
 	table = stagedInjectionTbl, stagedInjectionMap, "Staged Injection %", 1
 		xyLabels = "RPM", ""
 		xBins = injectorStagingRpmBins, RPMValue
-		yBins = injectorStagingLoadBins, fuelingLoad
+		yBins = injectorStagingLoadBins, afrTableYAxis
 		zBins = injectorStagingTable
 
 	table = tcuSolenoidTableTbl,  tcuSolenoidTableMap,  "Solenoids Active By Gear",	1