diff --git a/main.py b/main.py
index f6a375e..513f076 100644
--- a/main.py
+++ b/main.py
@@ -34,6 +34,10 @@ parser.add_argument('--list-models', action='store_true',
 
 args = parser.parse_args()
 
+if args.model == "baseline":
+    print("Model name 'baseline' not allowed")
+    exit()
+
 config = {
     'model': args.model,
     'episode_count': args.episode_count,