diff --git a/ts/metrics/metric_collector.py b/ts/metrics/metric_collector.py
index 9032d42e41..6a811d1bcc 100644
--- a/ts/metrics/metric_collector.py
+++ b/ts/metrics/metric_collector.py
@@ -15,7 +15,7 @@
     parser = argparse.ArgumentParser()
 
     parser.add_argument(
-        "--gpus",
+        "--gpu",
         action="store",
         help="number of GPUs",
         type=int
@@ -24,6 +24,6 @@
 
     logging.basicConfig(stream=sys.stdout, format="%(message)s", level=logging.INFO)
 
-    system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpus)
+    system_metrics.collect_all(sys.modules['ts.metrics.system_metrics'], arguments.gpu)
 
     check_process_mem_usage(sys.stdin)