diff --git a/Scripts/gap_threshold_selection.py b/Scripts/gap_threshold_selection.py
index e976bd53612453078b25ab843d2114f52740d2a7..b1e00b13e4cdd26fe636cf2d97906080516bc717 100644
--- a/Scripts/gap_threshold_selection.py
+++ b/Scripts/gap_threshold_selection.py
@@ -28,7 +28,7 @@ def find_gaps(col, m):
     return gaps
 
 # Get the input file as an argument (collinear file)
-input_file = sys.argv[1]
+input_file = sys.argv[sys.argv.index('-i') + 1]
 
 # Convert the collinear file to a dataframe
 C_df_csv = pd.read_csv(input_file, sep=' ', header=None)