diff --git a/helpers/visceral_make_splits.py b/helpers/visceral_make_splits.py
index 1cbe3f56f0bf101555265ad22743dd64a1c45092..970f4f042492e99d9189c00dd561fbb97a0d90d2 100644
--- a/helpers/visceral_make_splits.py
+++ b/helpers/visceral_make_splits.py
@@ -68,7 +68,7 @@ def save_to_json(
     train_files: list[str],
     test_files: list[str],
     validation_files: list[str],
-    output_file: str,
+    output_file: pathlib.Path,
     organ_1_id: str,
 ):
     data = {
@@ -84,7 +84,7 @@ def save_to_json(
         ],
     }
 
-    with pathlib.Path(output_file).open("w") as json_file:
+    with output_file.open("w") as json_file:
         json.dump(data, json_file, indent=2)