diff --git a/server/settings/publish_plugins.py b/server/settings/publish_plugins.py
index c32d8d360c..028cf9fffa 100644
--- a/server/settings/publish_plugins.py
+++ b/server/settings/publish_plugins.py
@@ -918,7 +918,7 @@ class IntegrateHeroTemplateNameProfileModel(BaseSettingsModel):
     template_name: str = SettingsField(
         "",
         title="Template name",
-        enum_resolver=anatomy_template_items_enum(category="publish")
+        enum_resolver=anatomy_template_items_enum(category="hero")
     )
 
 
diff --git a/server/settings/tools.py b/server/settings/tools.py
index 28ceeedbe4..b003ef2244 100644
--- a/server/settings/tools.py
+++ b/server/settings/tools.py
@@ -314,7 +314,7 @@ class CustomStagingDirProfileModel(BaseSettingsModel):
     template_name: str = SettingsField(
         "",
         title="Template name",
-        enum_resolver=anatomy_template_items_enum(category="publish")
+        enum_resolver=anatomy_template_items_enum(category="staging")
     )