diff --git a/cli.py b/cli.py
index c44dd89..b7f8d82 100644
--- a/cli.py
+++ b/cli.py
@@ -13,7 +13,7 @@
 args = trader_cli.get_args()
 
 rewards = {"incremental-profit": IncrementalProfit, "weighted-unrealized-profit": WeightedUnrealizedProfit}
-reward_strategy = rewards[args.reward_strategy]
+reward_strategy = rewards[args.reward_strat]
 
 
 def run_optimize(args, logger):
diff --git a/lib/cli/RLTraderCLI.py b/lib/cli/RLTraderCLI.py
index 32974ea..d9966a9 100644
--- a/lib/cli/RLTraderCLI.py
+++ b/lib/cli/RLTraderCLI.py
@@ -24,7 +24,7 @@ def __init__(self):
 
         self.parser.add_argument("--data-provider", "-d", type=str, default="static")
         self.parser.add_argument("--input-data-path", "-n", type=str, default="data/input/coinbase-1h-btc-usd.csv")
-        self.parser.add_argument("--reward-strategy", "-r", type=str, default="incremental-profit")
+        self.parser.add_argument("--reward-strategy", "-r", type=str, default="incremental-profit", dest="reward_strat")
         self.parser.add_argument("--pair", "-p", type=str, default="BTC/USD")
         self.parser.add_argument("--debug", "-D", action='store_false')
         self.parser.add_argument('--mini-batches', type=int, default=1, help='Mini batches', dest='n_minibatches')
diff --git a/optimize.py b/optimize.py
index 27203a6..ec7fc81 100644
--- a/optimize.py
+++ b/optimize.py
@@ -23,4 +23,4 @@ def optimize_code(params):
     from lib.RLTrader import RLTrader
 
     trader = RLTrader(**params)
-    trader.train(test_trained_model=True, render_trained_model=True)
+    trader.train(test_trained_model=True, render_test_env=True, render_report=True)