diff --git a/src/main.py b/src/main.py
index 050fa30..04e563a 100644
--- a/src/main.py
+++ b/src/main.py
@@ -21,7 +21,7 @@ parser.add_argument("--test", action="store_true", default=False)
 parser.add_argument("--model_path", type=str, default="results/model.pt")
 parser.add_argument("--results_path", type=str, default="results/results.csv")
 parser.add_argument("--data_acc_path", type=str, default="./results/acc.csv")
-parser.add_argument("--build_id", type=int, default=0)
+parser.add_argument("--build_id", type=str, default="0")
 
 # HYPER PARAMETERS
 parser.add_argument("--batch", "-b", type=int, default=2)
@@ -86,6 +86,6 @@ if __name__ == "__main__":
         utils.save_results(labels=test_data["label"], results=results, file_path=args.results_path)
         utils.save_data_to_csv(
             file_path=args.data_acc_path,
-            build_id=args.build_id,
+            build_id=int(args.build_id),
             data=accuracy,
         )