diff --git a/ArduSub/system.cpp b/ArduSub/system.cpp
index e2ce86de25..0563b5c77a 100644
--- a/ArduSub/system.cpp
+++ b/ArduSub/system.cpp
@@ -177,7 +177,7 @@ void Sub::startup_INS_ground()
 {
     // initialise ahrs (may push imu calibration into the mpu6000 if using that device).
     ahrs.init();
-    ahrs.set_vehicle_class(AHRS_VEHICLE_SUBMARINE);
+    ahrs.set_vehicle_class(AP_AHRS::VehicleClass::SUBMARINE);
 
     // Warm up and calibrate gyro offsets
     ins.init(scheduler.get_loop_rate_hz());