Fix

colony/core/roles/manager.py

@@ -6,9 +6,10 @@ from colony.core.logger.logging import get_logger
 
 
 class RoleManager:
-    def __init__(self, network, role_registry):
+    def __init__(self, network, role_registry, state_manager: StateManager):
         self.network = network
         self.role_registry = role_registry
+        self.state_manager = state_manager
 
         self.logger = get_logger(self.__class__.__name__)
 
@@ -17,7 +18,7 @@ class RoleManager:
         context = RoleContext(self.network, self)
 
         roles = {}
-        for role_name in StateManager().get(StateManager.Data.Roles):
+        for role_name in self.state_manager.get(StateManager.Data.Roles):
             role_cls = self.role_registry.get_role_class_by_name(role_name)
             self.logger.debug(f"Creating instance of role {role_name}. ")
             role_instance = role_cls(context=context)
@@ -55,7 +56,7 @@ class RoleManager:
             role_cls = self.role_registry.get_role_class_by_name(role_name)
             current_roles[role_name] = role_cls(context=context)
 
-        StateManager().set(StateManager.Data.Roles, list(current_roles.keys()))
+        self.state_manager.set(StateManager.Data.Roles, list(current_roles.keys()))
 
         self.network.self_node.roles = list(current_roles.keys())
         self.network.self_node.set_role_instances(current_roles)