diff --git a/DocumentUnderstanding/VGT/object_detection/create_grid_input.py b/DocumentUnderstanding/VGT/object_detection/create_grid_input.py
index f04d85e..a655e9a 100644
--- a/DocumentUnderstanding/VGT/object_detection/create_grid_input.py
+++ b/DocumentUnderstanding/VGT/object_detection/create_grid_input.py
@@ -212,4 +212,4 @@ def select_tokenizer(tokenizer):
     for page in range(len(word_grid)):
 
         grid = create_grid_dict(tokenizer, word_grid[page])
-        save_pkl_file(grid, args.output, f"page_{page}", page, args.model)
+        save_pkl_file(grid, args.output, f"page_{page}", args.model)