diff --git a/lib/models.py b/lib/models.py
index 36ae45f..f69955f 100644
--- a/lib/models.py
+++ b/lib/models.py
@@ -48,7 +48,7 @@ def _prepare_base_model(self, base_model_name):
             self.base_model = eval(base_model_name)(pretrained=self.pretrained,
                                     feat=True, pretrained_model=base_model_dict)
         else:
-            raise ValueError('Unknown base model: {}'.format(base_model))
+            raise ValueError('Unknown base model: {}'.format(base_model_name))
 
         # classifier: (dropout) + fc
         if self.dropout == 0: