X-Git-Url: https://git.armaanb.net/?p=lightcards.git;a=blobdiff_plain;f=lightcards%2Flightcards.py;h=62019a52b6078f07d3999e27cfda033f906c70e0;hp=775b13d24afaea722dc63296ef995e366a16c794;hb=589f6fda8e54d5b5ed0ae3b1c25aa7357974ddd3;hpb=7121c7107bcb4b3ef7a78878abf02c2e68354bf6

diff --git a/lightcards/lightcards.py b/lightcards/lightcards.py
index 775b13d..62019a5 100755
--- a/lightcards/lightcards.py
+++ b/lightcards/lightcards.py
@@ -4,6 +4,7 @@
 
 import argparse
 from curses import wrapper
+import sys
 
 from . import display, parse
 from .deck import Status
@@ -26,7 +27,7 @@ def show(stack, headers):
     wrapper(display.get_key, stack, headers, idx)
 
 
-def main():
+def main(args=sys.argv):
     args = parse_args()
     if args.version:
         print("0.0.0")