diff --git a/workflow/tests/test_pool_and_psuedoreplicate.py b/workflow/tests/test_pool_and_psuedoreplicate.py
index 7f8945b489200ca4202e76b80598127446c4643c..d600fd21261c92a885092fa3f4e5f79944a4235a 100644
--- a/workflow/tests/test_pool_and_psuedoreplicate.py
+++ b/workflow/tests/test_pool_and_psuedoreplicate.py
@@ -75,7 +75,7 @@ def test_single_rep(design_experiment_2):
     single_rep = pool_and_psuedoreplicate.generate_design('false', 1.2, design_experiment_2, cwd, 1, 1)
     assert single_rep.shape[0] == 4
     assert len(single_rep['control_tag_align'].unique()) == 1
-    assert single_rep['control_tag_align'].unique()[0] == 'B_1.tagAlign.gz'
+    assert 'pool_control.tagAlign.gz' in single_rep['control_tag_align'].unique()[0]
 
 
 @pytest.mark.unit
@@ -85,7 +85,7 @@ def test_single_control(design_experiment_3):
     shutil.copy(test_design_path + 'B_1.bedpe.gz', cwd)
     shutil.copy(test_design_path + 'A_1.tagAlign.gz', cwd)
     single_control = pool_and_psuedoreplicate.generate_design('true', 1.2, design_experiment_3, cwd, 1, 1)
-    assert single_control['control_tag_align'].unique() == 'B_1.tagAlign.gz'
+    assert 'pool_control.tagAlign.gz' in single_control['control_tag_align'].unique()[0] 
 
 
 @pytest.mark.singleend