diff --git a/yolov5/detect.py b/yolov5/detect.py
index 34beba00814fbc0e68f5b3b5c307d4d49e6fce70..cd69d9ca7ea30a23ae36a9e0077b980a1e29d23e 100755
--- a/yolov5/detect.py
+++ b/yolov5/detect.py
@@ -98,7 +98,7 @@ def run(
     screenshot = source.lower().startswith('screen')
     if is_url and is_file:
         source = check_file(source)  # download
-    if not args.name:
+    if not name:
         project_name = input('Please enter the name of your project : ')
     else:
         project_name = args.name
diff --git a/yolov5/train.py b/yolov5/train.py
index e18d4ea2227a352c365e1d19effbf9f06799e78c..c1e109ffc16e09bdbf9a477ca3c94e7b13e76bfb 100755
--- a/yolov5/train.py
+++ b/yolov5/train.py
@@ -512,7 +512,7 @@ def main(opt, callbacks=Callbacks()):
             opt.exist_ok, opt.resume = opt.resume, False  # pass resume to exist_ok and disable resume
         if opt.name == 'cfg':
             opt.name = Path(opt.cfg).stem  # use model.yaml as name
-        if not args.name:
+        if not opt.name:
             project_name = input('Please enter the name of your project : ')
         else:
             project_name = args.name