diff --git a/workflow/tests/test_pool_and_psuedoreplicate.py b/workflow/tests/test_pool_and_psuedoreplicate.py
index d600fd21261c92a885092fa3f4e5f79944a4235a..8c747056e22f154aaff28026dce9443d4bbd056e 100644
--- a/workflow/tests/test_pool_and_psuedoreplicate.py
+++ b/workflow/tests/test_pool_and_psuedoreplicate.py
@@ -74,8 +74,8 @@ def test_single_rep(design_experiment_2):
     shutil.copy(test_design_path + 'B_1.tagAlign.gz', cwd)
     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 'pool_control.tagAlign.gz' in single_rep['control_tag_align'].unique()[0]
+    assert len(single_rep['control_tag_align'].unique()) == 2
+    assert 'pool_control.tagAlign.gz' in single_rep['control_tag_align'].unique()[1]
 
 
 @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 'pool_control.tagAlign.gz' in single_control['control_tag_align'].unique()[0] 
+    assert 'B_1.tagAlign.gz' in single_control['control_tag_align'].unique()[1]
 
 
 @pytest.mark.singleend