diff --git a/keystore2/src/enforcements.rs b/keystore2/src/enforcements.rs
index 43147e8..04f26e9 100644
--- a/keystore2/src/enforcements.rs
+++ b/keystore2/src/enforcements.rs
@@ -603,7 +603,7 @@
             }
         }
 
-        if android_security_flags::fix_unlocked_device_required_keys() {
+        if android_security_flags::fix_unlocked_device_required_keys_v2() {
             let (hat, state) = if user_secure_ids.is_empty() {
                 (None, DeferredAuthState::NoAuthRequired)
             } else if let Some(key_time_out) = key_time_out {