diff --git a/RWKV-v4neo/train.py b/RWKV-v4neo/train.py
index 2b2d3e8..fd90b10 100644
--- a/RWKV-v4neo/train.py
+++ b/RWKV-v4neo/train.py
@@ -100,6 +100,7 @@ if __name__ == "__main__":
     parser.add_argument("--my_pos_emb", default=0, type=int)
     parser.add_argument("--load_partial", default=0, type=int)
     parser.add_argument("--magic_prime", default=0, type=int)
+    parser.add_argument("--my_testing", default=0, type=int)
 
     parser = Trainer.add_argparse_args(parser)
     args = parser.parse_args()