diff --git a/workflow/tests/test_call_peaks_macs.py b/workflow/tests/test_call_peaks_macs.py
index ebc8bb13110d18cd565bf5a68e46ad9f0dcc209d..e9a876dd33e6700f63ff413860e4c686ff64bbd6 100644
--- a/workflow/tests/test_call_peaks_macs.py
+++ b/workflow/tests/test_call_peaks_macs.py
@@ -1,11 +1,8 @@
 #!/usr/bin/env python3
 
 import pytest
-import pandas as pd
-from io import StringIO
 import os
 import utils
-import call_peaks_macs
 
 test_output_path = os.path.dirname(os.path.abspath(__file__)) + \
                 '/../output/callPeaksMACS/'
diff --git a/workflow/tests/test_overlap_peaks.py b/workflow/tests/test_overlap_peaks.py
index 6305d3793c0100b9449771c164ec0e3e5b08071d..524dd639c57e4a19f283ec5a1b3d4768e1922c9f 100644
--- a/workflow/tests/test_overlap_peaks.py
+++ b/workflow/tests/test_overlap_peaks.py
@@ -4,6 +4,7 @@ import pytest
 import pandas as pd
 from io import StringIO
 import os
+import utils
 import overlap_peaks
 
 test_output_path = os.path.dirname(os.path.abspath(__file__)) + \
@@ -26,7 +27,7 @@ def design_diff():
 
 def test_check_update_design(design_diff):
     new_design = overlap_peaks.update_design(design_diff)
-    assert df.shape[0] == 2
+    assert new_design.shape[0] == 2
     assert new_design.loc[0, 'control_bam_reads'] == "B_1.bam"
     assert new_design.loc[0, 'peak_caller'] == "bed"