diff --git a/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py b/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py
index e667113..cc8d695 100644
--- a/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py
+++ b/fairpyx/algorithms/heterogeneous_matroid_constraints_algorithms.py
@@ -21,8 +21,7 @@
 import base64
 
 logger = logging.getLogger(__name__)
-# Create a string stream to capture logs
-log_stream = io.StringIO()
+
 
 def per_category_round_robin(alloc: AllocationBuilder, item_categories: dict[str,list], agent_category_capacities: dict[str,dict[str,int]],
                              initial_agent_order: list,callback:callable=None):
@@ -1272,13 +1271,16 @@ def helper_generate_bipartite_graph_base64(graph,iteration:int,category:str):
     return base64.b64encode(img_bytes.read()).decode('utf-8')
 
 def helper_configure_logger():
+    # Create a string stream to capture logs
+    log_stream = io.StringIO()
     stream_handler = logging.StreamHandler(log_stream)
     formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
     logger.addHandler(stream_handler)
     stream_handler.setFormatter(formatter)
     logger.setLevel(logging.DEBUG)
+    return log_stream
 
-def helper_get_logs():
+def helper_get_logs(log_stream):
     return log_stream.getvalue()
 
 if __name__ == "__main__":