diff --git a/openpype/hosts/testhost/plugins/publish/collect_context.py b/openpype/hosts/testhost/plugins/publish/collect_context.py
index bbb8477cdf..0ab98fb84b 100644
--- a/openpype/hosts/testhost/plugins/publish/collect_context.py
+++ b/openpype/hosts/testhost/plugins/publish/collect_context.py
@@ -19,7 +19,7 @@ class CollectContextDataTestHost(
     hosts = ["testhost"]
 
     @classmethod
-    def get_instance_attr_defs(cls):
+    def get_attribute_defs(cls):
         return [
             attribute_definitions.BoolDef(
                 "test_bool",
diff --git a/openpype/hosts/testhost/plugins/publish/collect_instance_1.py b/openpype/hosts/testhost/plugins/publish/collect_instance_1.py
index 979ab83f11..3c035eccb6 100644
--- a/openpype/hosts/testhost/plugins/publish/collect_instance_1.py
+++ b/openpype/hosts/testhost/plugins/publish/collect_instance_1.py
@@ -20,7 +20,7 @@ class CollectInstanceOneTestHost(
     hosts = ["testhost"]
 
     @classmethod
-    def get_instance_attr_defs(cls):
+    def get_attribute_defs(cls):
         return [
             attribute_definitions.NumberDef(
                 "version",