From 2289c465b53f281c4767f0e8d746845c1f01ba63 Mon Sep 17 00:00:00 2001
From: Matthew Kennedy <matthewkennedy@outlook.com>
Date: Tue, 28 Jul 2020 11:56:39 -0700
Subject: [PATCH] fixes

---
 unit_tests/engine_test_helper.cpp | 2 +-
 unit_tests/engine_test_helper.h   | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

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;