diff --git a/tic_tac_toe.py b/tic_tac_toe.py
index f664851..3a939a4 100644
--- a/tic_tac_toe.py
+++ b/tic_tac_toe.py
@@ -60,10 +60,10 @@ def play(player_provider: PlayerProvider, k_max=10000, with_debug=False):
 if __name__ == '__main__':
 	parser = argparse.ArgumentParser()
 	parser.add_argument("--verbose", default=False, action='store_true', help="Be verbose")
-	parser.add_argument("--train", default=False, help="Train before play")
+	parser.add_argument("--train", default=False, action=argparse.BooleanOptionalAction, help="Train before play")
 	parser.add_argument("--num", default=5, type=int, help="Number of games to play")
-	parser.add_argument("--human", default=False, help="X-player is human")
-	parser.add_argument("--load", default=True, help="Load experience from previous trainings")
+	parser.add_argument("--human", default=False, action=argparse.BooleanOptionalAction, help="X-player is human")
+	parser.add_argument("--load", default=True, action=argparse.BooleanOptionalAction, help="Load experience from previous trainings")
 	args = parser.parse_args()
 
 	if args.load: