diff --git a/app/BUILD.gn b/app/BUILD.gn
index 7a6e5e42d6b..3ae44e1cb4a 100644
--- a/app/BUILD.gn
+++ b/app/BUILD.gn
@@ -192,7 +192,7 @@ source_set("unit_tests") {
       "//chrome/app:command_ids",
       "//chrome/browser/ui/toasts",
       "//chrome/common:chrome_features",
-      "//components/legion",
+      "//components/private_ai",
       "//components/translate/core/common",
       "//extensions/common",
       "//ui/accessibility:ax_base",
diff --git a/app/DEPS b/app/DEPS
index c25c3aed619..051bbcbc828 100644
--- a/app/DEPS
+++ b/app/DEPS
@@ -36,7 +36,7 @@ include_rules += [
   "+components/history_clusters/core/features.h",
   "+components/history_clusters/core/on_device_clustering_features.h",
   "+components/history_embeddings/core/history_embeddings_features.h",
-  "+components/legion/features.h",
+  "+components/private_ai/features.h",
   "+components/lens/lens_features.h",
   "+components/manta/features.h",
   "+components/metrics/metrics_features.h",
diff --git a/app/feature_defaults_unittest.cc b/app/feature_defaults_unittest.cc
index 2aaa1f0e1a7..6a13bb57f7f 100644
--- a/app/feature_defaults_unittest.cc
+++ b/app/feature_defaults_unittest.cc
@@ -29,7 +29,6 @@
 #include "components/history_clusters/core/features.h"
 #include "components/history_clusters/core/on_device_clustering_features.h"
 #include "components/history_embeddings/core/history_embeddings_features.h"
-#include "components/legion/features.h"
 #include "components/lens/lens_features.h"
 #include "components/manta/features.h"
 #include "components/metrics/metrics_features.h"
@@ -44,6 +43,7 @@
 #include "components/permissions/features.h"
 #include "components/plus_addresses/core/common/features.h"
 #include "components/privacy_sandbox/privacy_sandbox_features.h"
+#include "components/private_ai/features.h"
 #include "components/safe_browsing/core/common/features.h"
 #include "components/search/ntp_features.h"
 #include "components/segmentation_platform/public/features.h"
diff --git a/chromium_src/components/legion/features.cc b/chromium_src/components/private_ai/features.cc
similarity index 91%
rename from chromium_src/components/legion/features.cc
rename to chromium_src/components/private_ai/features.cc
index a06127ed97d..c2e9e323283 100644
--- a/chromium_src/components/legion/features.cc
+++ b/chromium_src/components/private_ai/features.cc
@@ -5,7 +5,7 @@
 
 #include "base/feature_override.h"
 
-#include <components/legion/features.cc>
+#include <components/private_ai/features.cc>
 
 namespace private_ai {