diff --git a/unit_tests/engine_test_helper.cpp b/unit_tests/engine_test_helper.cpp
index ad223147b6..260496a530 100644
--- a/unit_tests/engine_test_helper.cpp
+++ b/unit_tests/engine_test_helper.cpp
@@ -85,7 +85,7 @@ EngineTestHelper::EngineTestHelper(engine_type_e engineType, configuration_callb
 	engine->periodicSlowCallback(PASS_ENGINE_PARAMETER_SIGNATURE);
 
 	// Setup running in mock airmass mode
-	engineConfiguration->injectionMode = LM_MOCK;
+	engineConfiguration->fuelAlgorithm = LM_MOCK;
 	engine->mockAirmassModel = &mockAirmass;
 }
 
diff --git a/unit_tests/engine_test_helper.h b/unit_tests/engine_test_helper.h
index 799254b36f..a4ff83f055 100644
--- a/unit_tests/engine_test_helper.h
+++ b/unit_tests/engine_test_helper.h
@@ -13,6 +13,7 @@
 #include "main_trigger_callback.h"
 #include "unit_test_framework.h"
 #include "sensor.h"
+#include "mocks.h"
 
 extern EnginePins enginePins;