diff --git a/unit_tests/mocks.h b/unit_tests/mocks.h
index dcaad8ebb4..71c0bc3530 100644
--- a/unit_tests/mocks.h
+++ b/unit_tests/mocks.h
@@ -52,3 +52,8 @@ public:
 	MOCK_METHOD(void, scheduleByTimestampNt, (scheduling_s *scheduling, efitime_t timeUs, action_s action), (override));
 	MOCK_METHOD(void, scheduleForLater, (scheduling_s *scheduling, int delayUs, action_s action), (override));
 };
+
+class MockAirmass : public AirmassModelBase {
+public:
+	MOCK_METHOD(AirmassResult, getAirmass, (int rpm), (override));
+};