refactor code

1 files changed, 2 insertions, 3 deletions

diff --git a/wqflask/tests/wqflask/marker_regression/test_gemma_mapping.py b/wqflask/tests/wqflask/marker_regression/test_gemma_mapping.py
index a44ccb00..af20c1b8 100644
--- a/wqflask/tests/wqflask/marker_regression/test_gemma_mapping.py
+++ b/wqflask/tests/wqflask/marker_regression/test_gemma_mapping.py
@@ -28,13 +28,12 @@ class TestGemmaMapping(unittest.TestCase):
         dataset = AttributeSetter(
             {"group": AttributeSetter({"genofile": "genofile.geno"})})
 
-        output_files = "file1"
-        use_loco = False
+        output_file = "file1"
         mock_parse_loco.return_value = []
         this_trait = AttributeSetter({"name": "t1"})
 
         result = run_gemma(this_trait=this_trait, this_dataset=dataset, samples=[], vals=[
-        ], covariates="", use_loco=True, first_run=False, output_files=output_files)
+        ], covariates="", use_loco=True, first_run=False, output_files=output_file)
 
         expected_results = ([], "file1")
         self.assertEqual(expected_results, result)