diff --git a/train.py b/train.py
index 5c1273ef..c9e0e8c2 100644
--- a/train.py
+++ b/train.py
@@ -83,7 +83,7 @@ def main(_argv):
             # freeze darknet
             darknet = model.get_layer('yolo_darknet')
             freeze_all(darknet)
-        elif FLAGS.mode == 'frozen':
+        elif FLAGS.transfer == 'frozen':
             # freeze everything
             freeze_all(model)
         else: