diff --git a/torchreid/components/branches.py b/torchreid/components/branches.py
index 09bfa74..5f72e85 100644
--- a/torchreid/components/branches.py
+++ b/torchreid/components/branches.py
@@ -26,7 +26,7 @@ def _get_middle_subbranch_for(self, backbone, args, last_branch_class):
 
     def _get_branches(self, backbone, args) -> list:
 
-        branch_names = frozenset(args['branches'])
+        branch_names = tuple(args['branches'])
         branch_list = []
 
         for branch_name in branch_names: