diff --git a/high_level_policy_main.py b/high_level_policy_main.py
index e22ad07158c0b25de5d695aee21cbbacefd47b6e..bf1d542c6ce9e3387ac0019204e2e2c2c19f071e 100644
--- a/high_level_policy_main.py
+++ b/high_level_policy_main.py
@@ -172,7 +172,7 @@ if __name__ == "__main__":
 
     if args.train:
         high_level_policy_training(nb_steps=args.nb_steps, load_weights=args.load_weights, save_path=args.save_file,
-                                   tensorboard=args.tensorboard, nb_episodes_for_test=args.nb_episodes_for_test,
+                                   tensorboard=args.tensorboard, nb_episodes_for_test=20,
                                    visualize=args.visualize)
 
     if args.test: