diff --git a/libraries/AP_BLHeli/AP_BLHeli.cpp b/libraries/AP_BLHeli/AP_BLHeli.cpp
index 55f6a38987..3d033863c9 100644
--- a/libraries/AP_BLHeli/AP_BLHeli.cpp
+++ b/libraries/AP_BLHeli/AP_BLHeli.cpp
@@ -1339,7 +1339,7 @@ void AP_BLHeli::init(void)
     // setting the digital mask changes the min/max PWM values
     // it's important that this is NOT done for non-digital channels as otherwise
     // PWM min can result in motors turning. set for individual overrides first
-    if (mask && otype >= AP_HAL::RCOutput::MODE_PWM_DSHOT150) {
+    if (mask && hal.rcout->is_dshot_protocol(otype)) {
         digital_mask = mask;
     }