diff --git a/tests/test_ref.py b/tests/test_ref.py
index 21d9f6a..a128c9d 100755
--- a/tests/test_ref.py
+++ b/tests/test_ref.py
@@ -329,7 +329,7 @@ def test_ref(self):
                 self.gtf_path, use_version=True, filter_func=mock.ANY
             )
             create_t2g_from_fasta.assert_called_once_with(
-                cdna_fasta_path, t2g_path
+                cdna_fasta_path, t2g_path, aa_flag=False
             )
             split_genomic_fasta_to_cdna.assert_called_once_with(
                 self.fasta_path,
@@ -446,7 +446,7 @@ def test_ref_override_k(self):
                 self.gtf_path, use_version=True, filter_func=mock.ANY
             )
             create_t2g_from_fasta.assert_called_once_with(
-                cdna_fasta_path, t2g_path
+                cdna_fasta_path, t2g_path, aa_flag=False
             )
             split_genomic_fasta_to_cdna.assert_called_once_with(
                 self.fasta_path,
@@ -502,7 +502,7 @@ def test_ref_exists(self):
             split_genomic_fasta_to_cdna.assert_not_called()
             concatenate_files.assert_not_called()
             create_t2g_from_fasta.assert_called_once_with(
-                cdna_fasta_path, t2g_path
+                cdna_fasta_path, t2g_path, aa_flag=False
             )
             kallisto_index.assert_called_once_with(
                 cdna_fasta_path, index_path, k=31
@@ -591,7 +591,7 @@ def test_ref_overwrite(self):
                 self.gtf_path, use_version=True, filter_func=mock.ANY
             )
             create_t2g_from_fasta.assert_called_once_with(
-                cdna_fasta_path, t2g_path
+                cdna_fasta_path, t2g_path, aa_flag=False
             )
             split_genomic_fasta_to_cdna.assert_called_once_with(
                 self.fasta_path,