diff --git a/main.py b/main.py
index 5db8dd3..55754fe 100644
--- a/main.py
+++ b/main.py
@@ -310,7 +310,7 @@ def main(args):
 
     criterion = LabelSmoothingCrossEntropy()
 
-    if mixup_active > 0.:
+    if mixup_active:
         # smoothing is handled with mixup label transform
         criterion = SoftTargetCrossEntropy()
     elif args.smoothing: