diff --git a/src/main.py b/src/main.py
index fadbe657f6d7ebe195ec23089078db390eecf6fd..5d3c3ea5f616ac166c88cf6c0534ec3614a14392 100644
--- a/src/main.py
+++ b/src/main.py
@@ -12,7 +12,7 @@ checkpoint = utility.checkpoint(args)
 
 def main():
     global model
-    if args.data_test == 'video':
+    if args.data_test == ['video']:
         from videotester import VideoTester
         model = model.Model(args, checkpoint)
         t = VideoTester(args, model, checkpoint)