diff --git a/pype/lib/__init__.py b/pype/lib/__init__.py
index 6620bd5a12..a303bf038d 100644
--- a/pype/lib/__init__.py
+++ b/pype/lib/__init__.py
@@ -36,7 +36,6 @@ from .lib_old import (
     get_last_version_from_path,
     get_subsets,
     get_linked_assets,
-    map_subsets_by_family,
     BuildWorkfile,
     ffprobe_streams,
     source_hash,
diff --git a/pype/lib/lib_old.py b/pype/lib/lib_old.py
index 421e320415..b384c3a06a 100644
--- a/pype/lib/lib_old.py
+++ b/pype/lib/lib_old.py
@@ -608,20 +608,6 @@ def get_linked_assets(asset_entity):
     return inputs
 
 
-def map_subsets_by_family(subsets):
-    subsets_by_family = collections.defaultdict(list)
-    for subset in subsets:
-        family = subset["data"].get("family")
-        if not family:
-            families = subset["data"].get("families")
-            if not families:
-                continue
-            family = families[0]
-
-        subsets_by_family[family].append(subset)
-    return subsets_by_family
-
-
 class BuildWorkfile:
     """Wrapper for build workfile process.
 
@@ -629,6 +615,20 @@ class BuildWorkfile:
     are host related, since each host has it's loaders.
     """
 
+    @staticmethod
+    def map_subsets_by_family(subsets):
+        subsets_by_family = collections.defaultdict(list)
+        for subset in subsets:
+            family = subset["data"].get("family")
+            if not family:
+                families = subset["data"].get("families")
+                if not families:
+                    continue
+                family = families[0]
+
+            subsets_by_family[family].append(subset)
+        return subsets_by_family
+
     def process(self):
         """Main method of this wrapper.
 
@@ -901,7 +901,7 @@ class BuildWorkfile:
         :rtype: dict
         """
         # Prepare subsets
-        subsets_by_family = map_subsets_by_family(subsets)
+        subsets_by_family = self.map_subsets_by_family(subsets)
 
         profiles_per_subset_id = {}
         for family, subsets in subsets_by_family.items():