diff --git a/libraries/AP_Arming/AP_Arming.cpp b/libraries/AP_Arming/AP_Arming.cpp
index 4da3789e56..44e98bf4b6 100644
--- a/libraries/AP_Arming/AP_Arming.cpp
+++ b/libraries/AP_Arming/AP_Arming.cpp
@@ -31,7 +31,7 @@
 #include <AP_AHRS/AP_AHRS.h>
 #include <AP_Baro/AP_Baro.h>
 #include <AP_RangeFinder/AP_RangeFinder.h>
-#include <AP_Generator/AP_Generator_RichenPower.h>
+#include <AP_Generator/AP_Generator.h>
 #include <AP_Terrain/AP_Terrain.h>
 #include <AP_ADSB/AP_ADSB.h>
 #include <AP_Scripting/AP_Scripting.h>
@@ -1090,7 +1090,7 @@ bool AP_Arming::aux_auth_checks(bool display_failure)
 bool AP_Arming::generator_checks(bool display_failure) const
 {
 #if GENERATOR_ENABLED
-    const AP_Generator_RichenPower *generator = AP::generator();
+    const AP_Generator *generator = AP::generator();
     if (generator == nullptr) {
         return true;
     }