diff --git a/firmware/controllers/algo/engine2.cpp b/firmware/controllers/algo/engine2.cpp
index 66a72ec1d4..3d8f2ef305 100644
--- a/firmware/controllers/algo/engine2.cpp
+++ b/firmware/controllers/algo/engine2.cpp
@@ -130,9 +130,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;
 
@@ -143,6 +141,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 b5cf4a83bc..6dd024e782 100644
--- a/firmware/tunerstudio/tunerstudio.template.ini
+++ b/firmware/tunerstudio/tunerstudio.template.ini
@@ -1198,7 +1198,7 @@ curve = 32Curve, "3-2 Shift Solenoid Percent by Speed"
 	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