diff --git a/ee/app/policies/ee/user_policy.rb b/ee/app/policies/ee/user_policy.rb
index 001bad4c7c566986db1cdebcd1373f5546dcd386..769ac26caf2bad33c495d89bc15b51ce08bd7a11 100644
--- a/ee/app/policies/ee/user_policy.rb
+++ b/ee/app/policies/ee/user_policy.rb
@@ -56,7 +56,7 @@ def private_profile?
     end
 
     def can_assign_default_duo_group?
-      return false unless ::Feature.enabled?(:ai_model_switching, user)
+      return false unless ::Feature.enabled?(:ai_default_duo_namespace_user, user)
 
       ::Gitlab::CurrentSettings.current_application_settings.duo_features_enabled
     end
diff --git a/ee/spec/policies/user_policy_spec.rb b/ee/spec/policies/user_policy_spec.rb
index b8bba45fa5c402a31aaca0f3ca2e6cb60478b489..7348b812f04f6a2763994d277ccee1a0bcbc7545 100644
--- a/ee/spec/policies/user_policy_spec.rb
+++ b/ee/spec/policies/user_policy_spec.rb
@@ -254,7 +254,7 @@ def policy
   describe ':assign_default_duo_group' do
     using RSpec::Parameterized::TableSyntax
 
-    where(:ai_model_switching, :duo_features_enabled, :allowed?) do
+    where(:ai_default_duo_namespace_user, :duo_features_enabled, :allowed?) do
       false | true | false
       true | false | false
       false | false | false
@@ -263,7 +263,7 @@ def policy
 
     with_them do
       before do
-        stub_feature_flags(ai_model_switching: ai_model_switching)
+        stub_feature_flags(ai_default_duo_namespace_user: ai_default_duo_namespace_user)
         stub_application_setting(duo_features_enabled: duo_features_enabled)
       end