diff --git a/enhance.py b/enhance.py
index 0b9f12f..e1cfe5e 100644
--- a/enhance.py
+++ b/enhance.py
@@ -37,6 +37,7 @@ add_arg = parser.add_argument
 add_arg('files',                nargs='*', default=[])
 add_arg('--scales',             default=2, type=int,                help='How many times to perform 2x upsampling.')
 add_arg('--model',              default='medium', type=str,         help='Name of the neural network to load/save.')
+add_arg('--model',              default='small', type=str,          help='Name of the neural network to load/save.')
 add_arg('--train',              default=False, type=str,            help='File pattern to load for training.')
 add_arg('--batch-resolution',   default=192, type=int,              help='Resolution of images in training batch.')
 add_arg('--batch-size',         default=15, type=int,               help='Number of images per training batch.')