diff --git a/download_files_from_LGL_BW.py b/download_files_from_LGL_BW.py
index 9e4724c2ac8aec32647e6b38c48faf0da065010c..a9fcead00d2792797e7df0a2a0e83dc1b72e0464 100644
--- a/download_files_from_LGL_BW.py
+++ b/download_files_from_LGL_BW.py
@@ -241,14 +241,8 @@ Examples:
     return parser.parse_args()
 
 
-def main():
+def main(location_name: str, wkt_or_zipcode: str, download_only: bool = False, simstadt_folder: Path | None = None):
     """Main function to process arguments and run the download/extraction"""
-    args = parse_arguments()
-
-    location_name = args.name
-    wkt_or_zipcode = args.region
-    download_only = args.download_only
-    simstadt_folder = args.simstadt_folder
 
     # Validate location name
     if ' ' in location_name:
@@ -283,4 +277,6 @@ def main():
 
 
 if __name__ == '__main__':
-    main()
+    args = parse_arguments()
+    main(args.name, args.region, args.download_only, args.simstadt_folder)
+