Command line argument fix