use argparse version action to print allocbench version

1 files changed, 2 insertions, 5 deletions

diff --git a/bench.py b/bench.py
index 35a6b76..a10fe77 100755
--- a/bench.py
+++ b/bench.py
@@ -111,16 +111,13 @@ def main():
                         action='store_true')
     parser.add_argument("--version",
                         help="print version info and exit",
-                        action='store_true')
+                        action='version',
+                        version=f"allocbench {src.facter.allocbench_version()}")
 
     args = parser.parse_args()
     if args.license:
         print_license_and_exit()
 
-    if args.version:
-        print(src.facter.allocbench_version())
-        sys.exit(0)
-
     atexit.register(epilog)
 
     # Set global verbosity