diff --git a/action.yml b/action.yml
index a8804d0..991b17c 100644
--- a/action.yml
+++ b/action.yml
@@ -16,3 +16,7 @@ inputs:
 runs:
   using: 'docker'
   image: 'Dockerfile'
+  args:
+    - ${{ inputs.gtfs_file }}
+    - ${{ inputs.routes }}
+    - ${{ inputs.output_file }}
diff --git a/modify_gtfs.py b/modify_gtfs.py
index 8f1d7bc..906173f 100644
--- a/modify_gtfs.py
+++ b/modify_gtfs.py
@@ -46,6 +46,7 @@ def modify_routes(gtfs_file, routes, output_file=None):
 
     logging.info(f"Done!")
 
+
 if __name__ == "main":
     gtfs_file = sys.argv[1]
     routes = sys.argv[2]