diff --git a/sequence_extractor/cli.py b/sequence_extractor/cli.py
index c0e2e62219376cbb4d6f038ab8bde6c69c9e60ed..55c0393fc93670cfb345b41353efc74620934471 100644
--- a/sequence_extractor/cli.py
+++ b/sequence_extractor/cli.py
@@ -21,7 +21,7 @@ def main():
     LOG.info("sequence_extractor begins")
     exon_extraction_from_gtf()
     fasta_list = exon_concatenation(args.)
-    final_list = polyA_addition_to_fasta_list(fasta_list)
+    final_list = poly_a_addition_to_fasta_list(fasta_list)
     list_to_file(final_list,args.output_file_name)
     LOG.info("sequence_extractor ends")
 
diff --git a/tests/test_polyA.py b/tests/test_polyA.py
deleted file mode 100644
index ead1507e6b262dae6b705401d8b34ed61bbf9309..0000000000000000000000000000000000000000
--- a/tests/test_polyA.py
+++ /dev/null
@@ -1,3 +0,0 @@
-import pytest
-
-from polyA import poly
diff --git a/tests/test_poly_a.py b/tests/test_poly_a.py
new file mode 100644
index 0000000000000000000000000000000000000000..dc89197b9d69374bb1b165fdf37c088e13b9c800
--- /dev/null
+++ b/tests/test_poly_a.py
@@ -0,0 +1,3 @@
+import pytest
+
+from poly_a import poly_a_generator, poly_a_addition_to_fasta_list