diff --git a/libraries/AP_NavEKF/AP_NavEKF.cpp b/libraries/AP_NavEKF/AP_NavEKF.cpp
index c8c15de7aa..2c5ff8eb2d 100644
--- a/libraries/AP_NavEKF/AP_NavEKF.cpp
+++ b/libraries/AP_NavEKF/AP_NavEKF.cpp
@@ -25,11 +25,11 @@
   where the code is built. Note that this trick won't work for arduino
   builds on APM2, but NavEKF doesn't run on APM2, so that's OK
  */
-#if APM_BUILD_DIRECTORY == APM_BUILD_ArduCopter
+#if APM_BUILD_TYPE(APM_BUILD_ArduCopter)
 // copter defaults
 #define VELNE_NOISE_DEFAULT 0.30f
 
-#elif APM_BUILD_DIRECTORY == APM_BUILD_APMrover2
+#elif APM_BUILD_TYPE(APM_BUILD_APMrover2)
 // rover defaults
 #define VELNE_NOISE_DEFAULT 0.30f